from torch import nn
from torch.nn import functional as fn
from torch import Tensor
-from torch.autograd import Variable
from torch.nn import Module
import agtree2dot
return x
mlp = MLP(10, 20, 1)
-input = Variable(Tensor(100, 10).normal_())
-target = Variable(Tensor(100).normal_())
+input = Tensor(100, 10).normal_()
+target = Tensor(100, 1).normal_()
output = mlp(input)
criterion = nn.MSELoss()
loss = criterion(output, target)