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())
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