Присвойте значение тензору в соответствии с другим тензором
Я попытался использовать 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