bmul#
- radionets.evaluation.jet_angle.bmul(vec, mat, axis=0)[source]#
Expand vector for batchwise matrix multiplication.
- Parameters:
- vec2dtensor
vector for multiplication
- mat3dtensor
matrix for multiplication
- axisint, optional
batch axis, by default 0
- Returns
- ——-
- 3dtensor
Product of matrix multiplication. (bs, n, m)