oa::FnMatrix::bmm

bmm: per-batch matrix multiply, A[N,M,K] @ B[N,K,P] = out[N,M,P].

FnC++ public headers

Function Documentation

Matrix oa::FnMatrix::bmm(
	const Matrix & inA,
	const Matrix & inB
)

bmm: per-batch matrix multiply, A[N,M,K] @ B[N,K,P] = out[N,M,P].

inA

const Matrix &

—

inB

const Matrix &

—

Matrix

The declared return value.