def average_loss(self, h, a, t):
## print F.reshape(t, (-1, 1)).data
## print (h-F.reshape(t, (-1, 1))).data
self.loss = F.sum(abs(h - F.reshape(t, (-1,1))))
## self.loss = F.sqrt(F.sum(F.square(h - F.reshape(t, (-1,1)))))
self.loss /= self.n_patches
if self.n_images > 1:
h = F.split_axis(h, self.n_images, 0)
a = F.split_axis(a, self.n_images, 0)
else:
h, a = [h], [a]
self.y = h
self.a = a
评论列表
文章目录