Присвойте значение тензору в соответствии с другим тензором

Я попытался использовать tf.assign, чтобы присвоить значение моему тензору в соответствии с другим тензором, который имеет логическое значение. Код

self.labels_gt = self.labels_gt[self.gt_single_mask == False].assign(-1)

То, что я пытаюсь сделать, это просто, назначить -1 в self.labels_gt если соответствующий индекс в self.gt_single_mask является False, Но выдает ошибку

TypeError: Value passed to parameter 'begin' has DataType bool not in list of allowed values: int32, int64

0 ответов

Другие вопросы по тегам