标签： python numpy pytorch torch
我有一个名为84x84的{{1}} pytorch张量。我需要使用target和84x84组成的True布尔numpy数组对其进行遮罩。
84x84
target
True
当我做False时，出现错误target = target[mask]
False
target = target[mask]
令人惊讶的是，我仅在GPU上运行时才收到此错误。在CPU上运行时，一切正常。我该如何解决？