以下代码掩码良好。
mask = targets >= 0
targets = targets[mask]
然而,当我尝试使用两个条件进行掩蔽时,它会出现错误:RuntimeError: Boolean value of Tensor with more than one value is ambiguous
mask = (targets >= 0 and targets <= 5)
targets = targets[mask]
有办法做到这点吗?
(targets>=0) & (targets<=5)
,以在两个数组之间使用and
,详见我的回答。 - Akshay Sehgal