test_inv.py 文件源码

def test_identity_gpu(self):
        eye = cuda.to_gpu(_make_eye(self.x.shape))
        x = chainer.Variable(cuda.to_gpu(self.x))
        y = functions.batch_matmul(x, functions.batch_inv(x))
        gradient_check.assert_allclose(y.data, eye,
                                       **self.check_forward_options)