github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

revert creating Tensor from a list without numpy (#5041)

Browse Source 
the change was incomplete and broke creating Tensor from a list of np array
This commit is contained in:
chenyu
2024-06-18 17:31:22 -04:00
committed by GitHub
parent e2c5054bdd
commit 2b2488f2e2
3 changed files with 22 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_tensor.py
View File

@@ -320,6 +320,14 @@ class TestTinygrad(unittest.TestCase):
# data = data + [-x for x in data]
# np.testing.assert_allclose(Tensor(data, dtype=dtypes.int32).numpy(), np.array(data, dtype=np.int32))
def test_tensor_list_ndarray(self):
data = [np.array([1, 2, 3]), np.array([1, 2, 3]), np.array([1, 2, 3])]
np.testing.assert_equal(Tensor(data).numpy(), np.array(data))
data = [np.array([1.0, 2.0, 3.0]), np.array([1, 2, 3]), np.array([1, 2, 3])]
np.testing.assert_equal(Tensor(data).numpy(), np.array(data))
data = [np.array(1.0), np.array(2.0), np.array(3.0)]
np.testing.assert_equal(Tensor(data).numpy(), np.array(data))
def test_tensor_bytes(self):
data = b"abc123"
t = Tensor(data)