def testViewDataOnSqueeze(self): a = ones((1, 4, 1), chunk_size=2) b = squeeze(a, axis=0) b[:3] = 10 npa = np.ones((1, 4, 1)) npb = np.squeeze(npa, axis=0) npb[:3] = 10 np.testing.assert_array_equal(b.execute(), npb) np.testing.assert_array_equal(a.execute(), npa)
def testViewDataOnSqueeze(self): data = np.random.rand(1, 4, 1) a = tensor(data, chunk_size=2) b = squeeze(a, axis=0) b[:3] = 10 npa = data.copy() npb = np.squeeze(npa, axis=0) npb[:3] = 10 np.testing.assert_array_equal(b.execute(), npb) np.testing.assert_array_equal(a.execute(), npa)