Matrix Multiplication

Matrix multiplication of two tensors.

Usage

nv_matmul(lhs, rhs)

Arguments

lhs

(tensorish)

rhs

(tensorish)

Value

tensorish

Shapes

• lhs: (b1, ..., bk, m, n)

• rhs: (b1, ..., bk, n, p)

• output: (b1, ..., bk, m, p)