chainer_chemistry.functions.matmul¶

chainer_chemistry.functions.
matmul
(a, b, transa=False, transb=False)[source]¶ Computes the matrix multiplication of two arrays.
Parameters:  a (Variable) – The left operand of the matrix multiplication.
If
a
andb
are both 1D arrays,matmul
returns a dot product of vector a and vector b. If 2D arrays,matmul
returns matrix product ofa
andb
. If arrays’ dimension is larger than 2, they are treated as a stack of matrices residing in the last two indexes.matmul
returns a stack of each two arrays.a
andb
must have the same dimension.  b (Variable) – The right operand of the matrix multiplication.
Its array is treated as a matrix in the same way as
a
’s array.  transa (bool) – If
True
, each matrices ina
will be transposed. Ifa.ndim == 1
, do nothing.  transb (bool) – If
True
, each matrices inb
will be transposed. Ifb.ndim == 1
, do nothing.
Returns: The result of the matrix multiplication.
Return type: Variable
Example
>>> a = np.array([[1, 0], [0, 1]], 'f') >>> b = np.array([[4, 1], [2, 2]], 'f') >>> F.matmul(a, b).data array([[ 4., 1.], [ 2., 2.]], dtype=float32)
 a (Variable) – The left operand of the matrix multiplication.
If