本文简要介绍python语言中 torch.baddbmm
的用法。
用法:
torch.baddbmm(input, batch1, batch2, *, beta=1, alpha=1, out=None) → Tensor
beta(数字,可选的) -
input
( ) 的乘数alpha(数字,可选的) - ( ) 的乘数
out(Tensor,可选的) -输出张量。
对
batch1
和batch2
中的矩阵执行批次 matrix-matrix 乘积。input
添加到最终结果中。batch1
和batch2
必须是 3-D 张量,每个张量都包含相同数量的矩阵。如果
batch1
是 张量,batch2
是 张量,那么input
必须是可广播的 张量并且out
将是 张量。alpha
和beta
的含义与torch.addbmm()
中使用的比例因子相同。如果
beta
为0,则input
将被忽略,其中的nan
和inf
不会被传播。对于
FloatTensor
或DoubleTensor
类型的输入,参数beta
和alpha
必须是实数,否则它们应该是整数。该运算符支持 TensorFloat32。
例子:
>>> M = torch.randn(10, 3, 5) >>> batch1 = torch.randn(10, 3, 4) >>> batch2 = torch.randn(10, 4, 5) >>> torch.baddbmm(M, batch1, batch2).size() torch.Size([10, 3, 5])
参数:
关键字参数:
相关用法
- Python PyTorch backward用法及代码示例
- Python PyTorch batched_powerSGD_hook用法及代码示例
- Python PyTorch bincount用法及代码示例
- Python PyTorch bitwise_right_shift用法及代码示例
- Python PyTorch bernoulli用法及代码示例
- Python PyTorch bitwise_and用法及代码示例
- Python PyTorch bitwise_not用法及代码示例
- Python PyTorch binary_cross_entropy用法及代码示例
- Python PyTorch bitwise_xor用法及代码示例
- Python PyTorch binary_cross_entropy_with_logits用法及代码示例
- Python PyTorch bleu_score用法及代码示例
- Python PyTorch broadcast_tensors用法及代码示例
- Python PyTorch build_vocab_from_iterator用法及代码示例
- Python PyTorch broadcast_object_list用法及代码示例
- Python PyTorch broadcast_shapes用法及代码示例
- Python PyTorch bitwise_or用法及代码示例
- Python PyTorch bitwise_left_shift用法及代码示例
- Python PyTorch bf16_compress_wrapper用法及代码示例
- Python PyTorch bmm用法及代码示例
- Python PyTorch broadcast_to用法及代码示例
- Python PyTorch bf16_compress_hook用法及代码示例
- Python PyTorch bucketize用法及代码示例
- Python PyTorch block_diag用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.baddbmm。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。