def test_format(self):
dense_tensor = torch.rand(3, 4, 5, 6)
sparse_tensor = to_sparse(dense_tensor, format="BXXC")
self.assertEqual(len(sparse_tensor), 3 * 4 * 5)
self.assertEqual(sparse_tensor.F.size(1), 6)
def test(self):
dense_tensor = torch.rand(3, 4, 5, 6)
sparse_tensor = to_sparse(dense_tensor)
self.assertEqual(len(sparse_tensor), 3 * 5 * 6)
self.assertEqual(sparse_tensor.F.size(1), 4)