def check_forward(self, x_data, use_cudnn=True):
x = chainer.Variable(x_data)
y = functions.sigmoid(x, use_cudnn=use_cudnn)
self.assertEqual(y.data.dtype, self.dtype)
y_expect = functions.sigmoid(chainer.Variable(self.x))
gradient_check.assert_allclose(
y_expect.data, y.data, **self.check_forward_options)
评论列表
文章目录