Method valid_axes

dali/test/python/operator_2/test_reduce.py:288–289

(self)

def test_reduce_invalid_axes():
    class Batch3DInvalidAxes(Batch3D):
        def valid_axes(self):  # Invalid axes
            return [-100, (100, 0)]

    batch_fn = Batch3DInvalidAxes(np.uint8)
    dali_reduce_fn, numpy_reduce_fn = reduce_fns["sum"]