I have a [x y z] 3-d tensor and a [y z] 2-d tensor, how can I easily use pytorch to get the production, liker [1 y z][y z] ,[2 y z][y z]…[x y z]*[y z], all this result?
It’s not clear to me what operation you are looking to perform. Can you explain a bit more, e.g., give the sizes of the 3d and 2d tensor, and what you mean by the * operator, is it matrix multiply or pointwise multiply?
Maybe you could use torch.matmul?