Ejemplo n.º 1
0
    def testOnesLike(self):
        t1 = tensor([[0, 0, 1], [1, 0, 0]], chunk_size=2).tosparse()
        t = ones_like(t1, dtype='f8')

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

        t.tiles()
        self.assertEqual(t.chunks[0].index, (0, 0))
        self.assertIsInstance(t.op, TensorOnesLike)
        self.assertTrue(t.chunks[0].issparse())
Ejemplo n.º 2
0
def test_ones_like():
    t1 = tensor([[0, 0, 1], [1, 0, 0]], chunk_size=2).tosparse()
    t = ones_like(t1, dtype='f8')

    assert isinstance(t, SparseTensor)
    assert isinstance(t.op, TensorOnesLike)
    assert t.issparse() is True
    assert t.op.gpu is False

    t = tile(t)
    assert t.chunks[0].index == (0, 0)
    assert isinstance(t.op, TensorOnesLike)
    assert t.chunks[0].issparse() is True