def test_sparse_getitem_two_dim_slice():
actual = dense[2:4, 1:3]
res = sparse_getitem(sparse, (slice(2, 4), slice(1, 3)))
assert torch.equal(actual, res.to_dense())
微信
公众号
扫码关注公众号