def test_cumprod(self):
x = torch.rand(100, 100)
res1 = torch.cumprod(x, 1)
res2 = torch.Tensor()
torch.cumprod(res2, x, 1)
self.assertEqual(res1, res2)
微信
公众号
扫码关注公众号