我尝试使用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