torch (version 0.8.1)

torch_bmm: Bmm

Description

Bmm

Usage

torch_bmm(self, mat2)

Arguments

self

(Tensor) the first batch of matrices to be multiplied

mat2

(Tensor) the second batch of matrices to be multiplied

bmm(input, mat2, out=NULL) -> Tensor

Performs a batch matrix-matrix product of matrices stored in input and mat2.

input and mat2 must be 3-D tensors each containing the same number of matrices.

If input is a $(b\times n\times m)$ tensor, mat2 is a $(b\times m\times p)$ tensor, out will be a $(b\times n\times p)$ tensor.

$${\text{out}}_i={\text{input}}_i@{\text{mat2}}_i$$

Examples

if (torch_is_installed()) {

input = torch_randn(c(10, 3, 4))
mat2 = torch_randn(c(10, 4, 5))
res = torch_bmm(input, mat2)
res
}