def check_forward(self, xs):
y = chainerrl.functions.sum_arrays(xs)
correct_y = sum(self.xs)
gradient_check.assert_allclose(correct_y, cuda.to_cpu(y.data))
微信
公众号
扫码关注公众号