add shape attribute

Author: FynnBe

bioimageio/core/tensor.py

@@ -196,6 +196,11 @@ def ndim(self):
         """Number of tensor dimensions."""
         return self._data.ndim
 
+    @property
+    def shape(self):
+        """Tuple of tensor axes lengths"""
+        return self._data.shape
+
     @property
     def shape_tuple(self):
         """Tuple of tensor axes lengths"""

tests/test_tensor.py

@@ -60,3 +60,8 @@ def test_crop_and_pad():
 def test_some_magic_ops():
     tensor = Tensor.from_numpy(np.random.rand(256, 256), dims=None)
     assert tensor + 2 == 2 + tensor
+
+
+def test_shape_attributes():
+    tensor = Tensor.from_numpy(np.random.rand(1, 2, 25, 26), dims=None)
+    assert tensor.shape_tuple == tensor.shape