matmul

Signature

nabla.matmul(arg0, arg1) -> nabla.core.array.Array

Description

Matrix multiplication with broadcasting support.

Examples

import nabla as nb

# Matrix multiplication
A = nb.array([[1, 2], [3, 4]])
B = nb.array([[5, 6], [7, 8]])
result = nb.matmul(A, B)
print(result)  # [[19, 22], [43, 50]]