Method __init__

dali/test/python/operator_2/test_reduce.py:64–75

(self, data_type)

class Batch2D(Batch):
    def __init__(self, data_type):
        super().__init__(data_type)
        self._data = [
            [
                np.array([[1, 0, 2], [3, 1, 4]], dtype=self._data_type),
                np.array([[5, 0, 6], [7, 0, 8]], dtype=self._data_type),
            ],
            [
                np.array([[13, 23, 22], [23, 21, 14]], dtype=self._data_type),
                np.array([[23, 3, 6], [7, 0, 20]], dtype=self._data_type),
            ],
        ]

    def valid_axes(self):
        return [None, (), 0, 1, (0, 1)]