Как сделать поэлементное умножение для двух 4D тензоров неравного размера в pytorch?

Question

У меня есть тензор A и тензор B.

Размер A = [2,64,56,56]

Размер B = [2,64,29,29]

Как я могу выполнить torch.mul(A,B)? Тензоры имеют неравные размеры.

RuntimeError: shape [2, 64, 56, 56] недопустим для ввода размера 107648

Answer 1

Вы можете ознакомиться с документацией здесь: https://pytorch.org/docs/stable/torch.html#torch.mul

Там вы можете прочитать:

Формы ввода и другие должны быть транслируемыми.

Вы можете прочитать о возможности трансляции здесь: https://pytorch.org/docs/stable/notes/broadcasting.html#broadcasting-semantics

Наконец, возможно, имеет смысл подумать о том, чего вы хотите достичь.Я не уверен, что вы хотели, чтобы такая операция выполнялась, поскольку для поэлементного умножения необходимо, чтобы тензоры имели одинаковую форму.