добавление 2-х тензоров с разными размерами

Question

У меня есть 2 тензора с размерностью, A = [64,155,300] и B = [64,155,100], когда я добавляю эти 2 тензора ie. C = A + B, я получаю эту ошибку ==> "RuntimeError: размер тензора a (300) должен соответствовать размеру тензора b (100) в не-синглтонном измерении 2"

. помогите как мне добавить вышеперечисленные тензоры? любая помощь будет оценена!

Answer 1

Как говорит ошибка, вы не можете добавить два тензора с несовпадающими формами, но при желании вы можете повторить свой третий димер тензора B, чтобы он мог совпадать с A, используя torch.Tensor.repeat try A + B.repeat(1,1,3)

>>> A.shape
torch.Size([64, 155, 300])

>>> B.shape
torch.Size([64, 155, 100])

>>> B = B.repeat(1,1,3)

>>> B.shape
torch.Size([64, 155, 300])

>>> C = A + B

>>> C.shape
torch.Size([64, 155, 300])