mm

[common]
fun mm(input: Tensor, other: Tensor, bias: Tensor? = null, activation: Activation? = null): Tensor

mm(input: Tensor, other: Tensor, bias: Tensor? = null, activation: Activation? = null): the mm operator. Matrix multiply of two rank-2 tensors, with an optional fused bias and activation epilogue. a [M, K] x b [K, N] -> [M, N]. Inputs must be rank-2 (use bmm for batched operands, matmul for the broadcasting general form).