def __call__(self, x, t):
h1 = F.relu(self.l1(x))
h2 = F.relu(self.l2(h1))
h = F.relu(self.l3(h2))
self.loss = F.mean_squared_error(h, t)
return self.loss
微信
公众号
扫码关注公众号