def testBaseExecute(self): with option_context({'eager_mode': True}): a_data = np.random.rand(10, 10) a = mt.tensor(a_data, chunk_size=3) np.testing.assert_array_equal(a.fetch(), a_data) r1 = a + 1 np.testing.assert_array_equal(r1.fetch(), a_data + 1) r2 = 2 * r1 np.testing.assert_array_equal(r2.fetch(), (a_data + 1) * 2) # test add with out b = mt.ones((10, 10), chunk_size=3) np.testing.assert_array_equal(b.fetch(), np.ones((10, 10))) mt.add(a, b, out=b) np.testing.assert_array_equal(b.fetch(), a_data + 1) # test tensor dot c_data1 = np.random.rand(10, 10) c_data2 = np.random.rand(10, 10) c1 = mt.tensor(c_data1, chunk_size=4) c2 = mt.tensor(c_data2, chunk_size=4) r3 = c1.dot(c2) np.testing.assert_array_almost_equal(r3.fetch(), c_data1.dot(c_data2))
def testViewDataOnTranspose(self): a = ones((10, 20), chunk_size=6) b = a.T add(b, 1, out=b) np.testing.assert_array_equal(b.execute(), np.ones((20, 10)) + 1) np.testing.assert_array_equal(a.execute(), np.ones((10, 20)) + 1)
def testViewDataOnTranspose(self): data = np.random.rand(10, 20) a = tensor(data, chunk_size=6) b = a.T add(b, 1, out=b) np.testing.assert_array_equal(b.execute(), data.T + 1) np.testing.assert_array_equal(a.execute(), data + 1)
def test_view_data_on_transpose(setup): data = np.random.rand(10, 20) a = tensor(data, chunk_size=6) b = a.T add(b, 1, out=b) np.testing.assert_array_equal(b.execute(), data.T + 1) np.testing.assert_array_equal(a.execute(), data + 1)
def testViewDataOnAtleast2d(self): a = atleast_2d(ones(10, chunk_size=5)) b = add(a[:, :5], 1, out=a[:, 5:]) npa = np.atleast_2d(np.ones(10)) npb = np.add(npa[:, :5], 1, out=npa[:, 5:]) np.testing.assert_array_equal(b.execute(), npb) np.testing.assert_array_equal(a.execute(), npa)
def testViewDataOnAtleast2d(self): data = np.random.rand(10) a = atleast_2d(tensor(data, chunk_size=5)) b = add(a[:, :5], 1, out=a[:, 5:]) npa = np.atleast_2d(data.copy()) npb = np.add(npa[:, :5], 1, out=npa[:, 5:]) np.testing.assert_array_equal(b.execute(), npb) np.testing.assert_array_equal(a.execute(), npa)
def testViewDataOnAtleast3d(self): a = atleast_3d(ones((10, 20), chunk_size=5)) b = a[:, :5, :10][0] c = add(b[:4], b[1:], out=a[0, 16:]) npa = np.atleast_3d(np.ones((10, 20))) npb = npa[:, :5, :10][0] npc = np.add(npb[:4], npb[1:], out=npa[0, 16:]) np.testing.assert_array_equal(c.execute(), npc) np.testing.assert_array_equal(b.execute(), npb) np.testing.assert_array_equal(a.execute(), npa)
def testViewDataOnAtleast3d(self): data = np.random.rand(10, 20) a = atleast_3d(tensor(data, chunk_size=5)) b = a[:, :5, :10][0] c = add(b[:4], b[1:], out=a[0, 16:]) npa = np.atleast_3d(data.copy()) npb = npa[:, :5, :10][0] npc = np.add(npb[:4], npb[1:], out=npa[0, 16:]) np.testing.assert_array_equal(c.execute(), npc) np.testing.assert_array_equal(b.execute(), npb) np.testing.assert_array_equal(a.execute(), npa)