Hi.

I have two tensors.

a.size() is torch.size([64, 3, 3, 3])

b.size() is torch.size([64])

I wanna a[0]*b[0], a[1]*b[1], … a[63]*b[63].

Should I use for loop to operate above calculations?

Hi.

I have two tensors.

a.size() is torch.size([64, 3, 3, 3])

b.size() is torch.size([64])

I wanna a[0]*b[0], a[1]*b[1], … a[63]*b[63].

Should I use for loop to operate above calculations?

Hi,

What do you mean by `a[0]*b[0]`

? Because a[0] is a tensor while b[0] is a number?

If you want to add the value of b[0] element-wise to a[0], then `a * b.view(64, 1, 1, 1).expand_as(a)`

.