Mm
Note
This function does not broadcast .
For broadcasting matrix products, see torch_matmul
.
mm(input, mat2, out=NULL) -> Tensor
Performs a matrix multiplication of the matrices input
and mat2
.
If input
is a \((n \times m)\) tensor, mat2
is a
\((m \times p)\) tensor, out
will be a \((n \times p)\) tensor.
Examples
if (torch_is_installed()) {
mat1 = torch_randn(c(2, 3))
mat2 = torch_randn(c(3, 3))
torch_mm(mat1, mat2)
}
#> torch_tensor
#> 1.2318 -1.1066 0.9638
#> 0.5778 -1.5903 0.9098
#> [ CPUFloatType{2,3} ]