tf.boolean_mask не принимает аргумент оси

Question

Вот мой код:

 44     scores = tf.boolean_mask(box_class_scores,filtering_mask,axis=-1)
 45     boxes = tf.boolean_mask(boxes,filtering_mask,axis=-1)
 46     classes = tf.boolean_mask(box_classes,filtering_mask,axis=-1)

Ошибка, я получаю:

Ошибка типа: boolean_mask () получил неожиданный аргумент ключевого слова 'axis'

tf.boolean_mask() не принимает аргумент оси, но является допустимым аргументом, как видно из документации: https://www.tensorflow.org/api_docs/python/tf/boolean_mask

Answer 1

Как указал @syltruong, проверьте, соответствует ли используемая вами версия Keras документации.Прямо сейчас в документации используется версия 2.2.4.

В примечании @Ganesh и @Sushant на самом деле вам не нужно использовать параметр оси для этого конкретного фрагмента кода в специализации глубокого обучения.