Ошибка в коде pytorch : mat1 and mat2 shapes cannot be multiplied (1x100 and 1x15)

Код брал из репозитория тут

вылезет ошибка в forward функции на линии x=self.fc1(x)

Ошибка звучит так mat1 and mat2 shapes cannot be multiplied (1x100 and 1x15) Подскажите что с этим сделать? Как устранить ошибку?


Ответы (2 шт):

Автор решения: A_Vaclav

Нужно почитать "Умножение матриц" Матрицы A и B могут быть перемножены, если они совместимы в том смысле, что число столбцов матрицы A равно числу строк B

В вашем случае поможет транспонирование одной из матриц... если это допустимо.

→ Ссылка
Автор решения: MIHAnik22

Пропустил строки:

x_train.unsqueeze_(1)
y_train.unsqueeze_(1);
→ Ссылка