Как я могу поэлементно 2 Тензора с различными измерениями с вещанием?

Question

У меня есть тензор с именем inputs с size torch.Size([20, 1, 161, 199]) и еще один mask с размером torch.Size([20, 1, 199]). Я хочу умножить их вместе.

Я пытался:

masked_inputs = inputs * mask[..., None]

, но получил ошибку:

RuntimeError: The size of tensor a (161) must match the size of tensor b (199) at non-singleton dimension 2

Я не совсем уверен, что делать?

Answer 1

Это сделал это:

masked_inputs = inputs * mask.unsqueeze(2)