import torch.nn as nn class MyLinearRegressor2(nn.Module): def __init__(self, vlen): super(MyLinearRegressor2, self).__init__() self.w = nn.Linear(vlen, 1, bias=False) def forward(self, x): return self.w(x).squeeze()