Size

Contents

Size#

brainstate.typing.Size#

Type for specifying array sizes and dimensions.

Can be a single integer for 1D sizes, or a sequence of integers for multi-dimensional shapes. Supports both Python integers and NumPy integer types for compatibility.

Examples

>>> # Single dimension
>>> size1: Size = 10
>>>
>>> # Multiple dimensions
>>> size2: Size = (3, 4, 5)
>>>
>>> # Using NumPy integers
>>> size3: Size = np.int32(8)
>>>
>>> # Mixed sequence
>>> size4: Size = [np.int64(2), 3, np.int32(4)]