I have a (dense) `Tensor`

`X`

of shape `(A, B, C)`

that I need to (tensor-)multiply with a sparse tensor `Y`

of shape `(C,C)`

along the last axis of the `X`

and first axis of `Y`

independently of the first two axes of `X`

to create a new `Tensor`

of shape `(A, B, C)`

. How do I appropriately broadcast `Y`

? I definitely need to avoid looping over the first two axes of `X`

.

2 Likes