class Mylinear(nn.Module):
def __init__(self, inp, outp):
super(MyLinear, self).__init__()
#require_grad = True
self.w = nn.Parameter(torch.randn(outp, inp))
self.b = nn.Parameter(torch.randn(outp))
def forward(self, x)
x = x @ self.w.t() + self.b
return x