OperatorNotAllowedInGraphError: using a `tf.Tensor` as a Python `bool` is not allowed: AutoGraph did convert this function

Question

I\'m trying to filter a tensorflow.dataset by its index:

    dataset = tf.data.Dataset.from_tensor_slices((sequences_matrix, label_data.astype(np