Exemplo n.º 1
0
def test_from_dense():
    t = fromdense(tensor([[0, 0, 1], [1, 0, 0]], chunk_size=2))

    assert isinstance(t, SparseTensor)
    assert isinstance(t.op, DenseToSparse)
    assert t.issparse() is True

    t = tile(t)
    assert t.chunks[0].index == (0, 0)
    assert isinstance(t.op, DenseToSparse)
Exemplo n.º 2
0
    def testFromDense(self):
        t = fromdense(tensor([[0, 0, 1], [1, 0, 0]], chunk_size=2))

        self.assertIsInstance(t, SparseTensor)
        self.assertIsInstance(t.op, DenseToSparse)
        self.assertTrue(t.issparse())

        t = t.tiles()
        self.assertEqual(t.chunks[0].index, (0, 0))
        self.assertIsInstance(t.op, DenseToSparse)