chainer.functions.batch_matmul

chainer.functions.batch_matmul(a, b, transa=False, transb=False)[source]

Computes the batch matrix multiplications of two sets of arrays.

Parameters
  • a (Variable or N-dimensional array) – The left operand of the batch matrix multiplications. A 2-D array of shape (B, N) is considered as B \(N \times 1\) matrices. A 3-D array of shape (B, M, N) is considered as B \(M \times N\) matrices.

  • b (Variable or N-dimensional array) – The right operand of the batch matrix multiplications. Its array is treated as matrices in the same way as a’s array.

  • transa (bool) – If True, transpose each matrix in a.

  • transb (bool) – If True, transpose each matrix in b.

Returns

The result of the batch matrix multiplications as a 3-D array.

Return type

Variable

Deprecated since version v3.0.0: batch_matmul is deprecated. Use matmul instead.