import tensorflow as tf

            current_data = provider.rotate_tensor_by_label(current_data, current_label, self.graph)
        elif self.num_angles == 32:
            current_data = provider.rotate_tensor_by_label_32(current_data, current_label, self.graph)
        elif self.num_angles == 54:
            current_data = provider.rotate_tensor_by_label_54(current_data, current_label, self.graph)
        elif self.num_angles: #sunflower distribution
            current_data = provider.rotate_point_by_label_n(current_data, current_label, self.graph, self.num_angles, use_tensor=True)
        else:
            raise(NotImplementedError())
        current_data = tf.convert_to_tensor(current_data)
        return current_data

    