✨ HasShape

Author: nstarman

src/array_api_typing/_array.py

@@ -106,13 +106,35 @@ def ndim(self) -> int:
         ...
 
 
+class HasShape(Protocol):
+    """Protocol for array classes that have a shape attribute."""
+
+    @property
+    def shape(self) -> tuple[int | None, ...]:
+        """Shape of the array.
+
+        Returns:
+            tuple[int | None, ...]: array dimensions. An array dimension must be None
+                if and only if a dimension is unknown.
+
+        Notes:
+            For array libraries having graph-based computational models, array
+            dimensions may be unknown due to data-dependent operations (e.g.,
+            boolean indexing; `A[:, B > 0]`) and thus cannot be statically
+            resolved without knowing array contents.
+
+        """
+        ...
+
+
 @docstring_setter(**_array_docstrings)
 class Array(
     # ------ Attributes -------
     HasDType[DType_co],
     HasDevice,
     HasMatrixTranspose,
     HasNDim,
+    HasShape,
     # ------ Methods -------
     HasArrayNamespace[NS_co],
     op.CanPosSelf,