def testChunkSameSize(self):
t1 = TensorBase(np.random.randint(0, 10, size=(4, 10)))
t2, t3 = t1.chunk(2, 0, same_size=True)
self.assertEqual(t2.shape(), t3.shape())
def testChunk(self):
t1 = TensorBase(np.random.randint(0, 10, size=(5, 10)))
t2, t3 = t1.chunk(2, 0)
self.assertNotEqual(t2.shape(), t3.shape())
