Слои PyTorch

Например есть класс. 

 class LinearRegressionModel(nn.Module):

 def __init__(self):

 super().__init__()

 

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

 return self.weights * x + self.bias 

 nn.Linear 

 Слой линейной регрессии, in_features - кол-во входящих переменных, out_features - кол-во исходящих значений. В функции Y = F(X) это размерность входного и получаемого тензоров.