def test_5d_shape(self): x, interp_orders = self.get_5d_input() rand_rotation_layer = RandomRotationLayer( min_angle=-10.0, max_angle=10.0) rand_rotation_layer.randomise() out = rand_rotation_layer(x, interp_orders)
def test_5d_shape(self): x, interp_orders = self.get_5d_input() rand_rotation_layer = RandomRotationLayer() rand_rotation_layer.init_uniform_angle((-10.0, 10.0)) rand_rotation_layer.randomise() out = rand_rotation_layer(x, interp_orders)