From 92a0f6ea0accc6178414de117f37659ee06388b8 Mon Sep 17 00:00:00 2001 From: ice Date: Sat, 31 Jan 2026 14:37:30 +0800 Subject: [PATCH 1/2] Update test_tensors.py --- python/tests/core/test_tensors.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/python/tests/core/test_tensors.py b/python/tests/core/test_tensors.py index 38242ee..63a1432 100644 --- a/python/tests/core/test_tensors.py +++ b/python/tests/core/test_tensors.py @@ -29,6 +29,25 @@ def test_tensor_creation(self): for shape in shapes: tensor = Tensor(shape=shape) self.assertEqual(tensor.shape, shape) + + def test_tensor_is_same_shape(self): + """测试 Tensor 是否相同形状""" + tensor1 = Tensor(dtype=DataType.FLOAT32, shape=[2, 3]) + tensor2 = Tensor(dtype=DataType.FLOAT32, shape=[2, 3]) + tensor3 = Tensor(dtype=DataType.FLOAT32, shape=[3, 2]) + self.assertTrue(tensor1.is_same_shape(tensor2)) + self.assertFalse(tensor1.is_same_shape(tensor3)) + + def test_tensor_properties(self): + """测试 Tensor Properties""" + tensor = Tensor(dtype=DataType.FLOAT32, shape=[2, 3]) + self.assertEqual(tensor.is_cuda(), tensor.device == "cuda") + self.assertEqual(type(tensor.num_elements), str) + self.assertEqual(type(tensor.num_bytes), str) + self.assertEqual(type(ndim), int) + self.assertEqual(type(shape), list[int]) + self.assertEqual(type(device), str) + self.assertEqual(type(dtype), DataType) def test_tensor_data_types(self): """测试不同数据类型""" From fd231de78b0483ad93ab8820474d807c0bee57df Mon Sep 17 00:00:00 2001 From: ice Date: Sat, 31 Jan 2026 14:37:49 +0800 Subject: [PATCH 2/2] Remove create method from Tensor class Removed the create method from the Tensor class. --- python/axono/core/tensor.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/python/axono/core/tensor.py b/python/axono/core/tensor.py index 469fc8a..3d30997 100644 --- a/python/axono/core/tensor.py +++ b/python/axono/core/tensor.py @@ -49,11 +49,6 @@ def __init__( def is_cuda(self): return self._tensor.is_cuda - @classmethod - def create(cls, dtype: DataType, shape: list[int]) -> "Tensor": - """Create a new tensor""" - return cls(dtype, shape) - @classmethod def create_like(cls, other: "Tensor") -> "Tensor": """Create a tensor with same shape and dtype as another"""