python collections.abc

python sequence

from collections.abc import Sequence
import numpy as np

class Tensor3D(Sequence):
    def __init__(self, data):
        self._data = np.array(data)
        
    def __getitem__(self, index):
        return self._data[index]
    
    def __len__(self):
        return len(self._data)
    
    def sum_along_axis(self, axis):
        return np.sum(self._data, axis=axis)

# Usage
t = Tensor3D([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(t[0])  # Access first 2D slice
print(t.sum_along_axis(0))  # Sum along first axis