matrix multiplication — tch_mm • torch

Performs matrix multiplication for 2 tensors.

tch_mm(x, mat2)

Arguments

x	tensor object
mat2	second tensor object

Examples

x <- tensor(matrix(runif(10), ncol = 5))
y <- tensor(matrix(runif(10), nrow = 5))
tch_mm(x, y)
#> tensor 
#>  1.4303  0.9955
#>  1.3815  0.8157
#> [ Variable[CPUFloatType]{2,2} ]