如何解决PyTorch 索引：选择索引的补充
x = torch.tensor([1,2,3,4,5])
idx = torch.tensor([0,4])
如果我想选择索引中不的所有元素，我可以这样manually define a Boolean mask：
mask = torch.ones_like(x)
mask[idx] = 0
x[mask]
有没有更优雅的方法来做到这一点？
即我可以直接传递索引而不是创建掩码的语法，例如类似：
x[~idx]
解决方法
您可能想尝试单行表达式：
x[np.setdiff1d(range(len(x)),idx)]
虽然看起来也不优雅:)
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
