mm

dragon.vm.torch.mm(
  input,
  mat2,
  transpose_a=False,
  transpose_b=False,
  out=None
)[source]

Compute matrix-matrix multiplication.

\[y = a \times b \]
Parameters:
  • input (dragon.vm.torch.Tensor) – The matrix \(a\).
  • mat2 (dragon.vm.torch.Tensor) – The matrix \(b\).
  • transpose_a (bool, optional, default=False) – True to transpose \(a\) before computation.
  • transpose_b (bool, optional, default=False) – True to transpose \(b\) before computation.
  • out (dragon.vm.torch.Tensor, optional) – The optional output.
Returns:

dragon.vm.torch.Tensor – The output tensor.