array_collapse_right function

array_collapse_right

Calculates the matrix product between an array and a vector.

array_collapse_right(A, B)

Arguments

• A: A 3-D array with shapes n x m x k.
• B: A matrix with shapes 1 x n.

Details

For an array A with shapes m x n x k and a vector B with shape m, this operations returns a matrix C, with shapes n x k, so that C[,i] = B