#python #pytorch #tensor #conv-neural-network
#python #pytorch #тензор #conv-нейронная сеть
Вопрос:
Я получаю эту ошибку кода, и я не уверен, как изменить мой тензор.
RuntimeError: Expected 3-dimensional input for 3-dimensional weight [32, 35, 2], but got 2-dimensional input of size [35, 64] instead
Комментарии:
1. какой у вас код?
Ответ №1:
Вы можете использовать reshape функцию из torch следующим образом:
a = torch.rand((35,64))
b = torch.reshape(a, (32,35,2))