Addbmm
Addbmm
torch_addbmm(self, batch1, batch2, beta = 1L, alpha = 1L)
self |
(Tensor) matrix to be added |
batch1 |
(Tensor) the first batch of matrices to be multiplied |
batch2 |
(Tensor) the second batch of matrices to be multiplied |
beta |
(Number, optional) multiplier for |
alpha |
(Number, optional) multiplier for |
Performs a batch matrix-matrix product of matrices stored
in batch1 and batch2,
with a reduced add step (all matrix multiplications get accumulated
along the first dimension).
input is added to the final result.
batch1 and batch2 must be 3-D tensors each containing the
same number of matrices.
If batch1 is a (b \times n \times m) tensor, batch2 is a
(b \times m \times p) tensor, input must be
broadcastable with a (n \times p) tensor
and out will be a (n \times p) tensor.
out = β\ \mbox{input} + α\ (∑_{i=0}^{b-1} \mbox{batch1}_i \mathbin{@} \mbox{batch2}_i)
For inputs of type FloatTensor or DoubleTensor, arguments beta and alpha
must be real numbers, otherwise they should be integers.
if (torch_is_installed()) {
M = torch_randn(c(3, 5))
batch1 = torch_randn(c(10, 3, 4))
batch2 = torch_randn(c(10, 4, 5))
torch_addbmm(M, batch1, batch2)
}Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.