How to calculate 3dtensor with vector?

1

I am a newbie using pytorch.
Does anybody know how to calculate a 14x14x2048 tensor multiplied by a 2048-dim verctor to get another 14x14x2048 tensor?
Is there any function in pytorch?

2

You could just multiply both tensors and let broadcasting do the work:

x = torch.randn(14, 14, 2048)
y = torch.ones(2048) * 2
z = x*y
3

thank you viry much!