def test_transpose_float(input_shape, expected_shape): img = np.random.uniform(low=0.0, high=1.0, size=input_shape).astype('float32') transposed = F.transpose(img) assert transposed.shape == expected_shape
def test_transpose(input_shape, expected_shape): img = np.random.randint(low=0, high=256, size=input_shape, dtype=np.uint8) transposed = F.transpose(img) assert transposed.shape == expected_shape