小编jos*_*hua的帖子

具有多个条件的掩码列表/张量?

以下代码可以很好地屏蔽

mask = targets >= 0
targets = targets[mask]
Run Code Online (Sandbox Code Playgroud)

但是,当我尝试用两个条件进行屏蔽时,它会给出错误RuntimeError: Boolean value of Tensor with more than one value is ambiguous

mask = (targets >= 0 and targets <= 5)
targets = targets[mask]
Run Code Online (Sandbox Code Playgroud)

有没有办法做到这一点?

python list mask pytorch tensor

5
推荐指数
1
解决办法
5168
查看次数

标签 统计

list ×1

mask ×1

python ×1

pytorch ×1

tensor ×1