aitech-eks-pub/wyk/pytorch-regression/my_linear_regressor2.py

11 lines
252 B
Python

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()