If I have a 3d tensor of size
T1 --> torch.Size([196, 14, 14])
and a 1d tensor of size
T2 --> torch.Size([196])
How do I multiply matrices of size [14, 14] in T1 by the corresponding scalar in T2?
e.g if T2[0] == 3, then the first multiplication would be
T1[0] * 3 , etc.
So the output should be with the original shape of T1.