Почему я должен не допустить данных от Torch.Size ([50]) до Torch.Size ([50, 1]) ⇐ Python

[Anonymous]

Я проходил курс глубокого обучения Pytorch от Freecodecamp, и сомнение: < /p>

Код: Выделить всё

weight = 0.7
bias = 0.3
start = 0
end = 1
step = 0.02

X = torch.arange(start, end, step).unsqueeze(dim=1)
y=weight*X + bias
X[:10], y[:10]
train_split=int(0.8*len(X))
X_train, y_train = X[:train_split], y[:train_split]
X_test, y_test=X[train_split:], y[train_split:]

Почему функция Unsqueze используется для , чтобы сделать тензор размера [50, 1], а не [50] ? Наставник говорил, что это вызовет ошибку, но я не знаю, почему происходит ошибка? < /p>
После попытки тренировать модель, я получаю эту ошибку: < /p>

Код: Выделить всё

class LinearRegressionModelv2(nn.Module):
def __init__(self):
super().__init__()
self.linear_layer = nn.Linear(in_features=1, out_features=1)

def forward(self, x: torch.Tensor) -> torch.Tensor:
return self.linear_layer(x)

torch.manual_seed(42)
model_v2 = LinearRegressionModelv2()
< /code>
y_prediction = model_v2(X_train)

indexerror: измерение из диапазона (ожидается, что он будет в диапазоне [-1, 0], но получил -2)

Подробнее здесь: https://stackoverflow.com/questions/794 ... h-size50-1