def test_getitem():
res = make_mul_lazy_var()[0][1, 1]
assert torch.norm(res.evaluate().data - (t1_t2_t3_eval + torch.ones(4))[1, 1]) < 1e-3
微信
公众号
扫码关注公众号