tns1 = 7 x 7 x 3
Se multiplicarmos as dimensões:
7 * 7 * 3 = 147
Ou seja, o tns1 possui 147 elementos, assim como o tns2, que tem dimensão:
tns2 = 147 x 1
Com base na conclusão que cheguei acima descrevo o seguinte código à baixo:
import torch
tns1 = torch.randn(7, 7, 3)
tns2 = torch.randn(147, 1)
tns1 = tns1.reshape(147, 1)
resultado = tns1 + tns2
print(tns1.size())
print(tns2.size())
print(resultado.size())
a saída fica assim:
torch.Size([147, 1])
torch.Size([147, 1])
torch.Size([147, 1])
Conte com o apoio da comunidade Alura na sua jornada. Abraços e bons estudos!