x = [[1.0],[1.0]] W = [[1.0,1.0],[1.0,2.0]] input = torch.tensor(x) weights = torch.tensor(W, requires_grad=True) result = weights @ input lresult = result.tolist()