)
tensor([0, 0, 0, 1, 1], dtype=torch.uint8)
tensor([3, 4])
tensor([4, 5])
补充知识:pytorch tensor筛选满足条件的行或列...(使用与或)
我就废话不多说了,大家还是直接看代码吧~
import torch
x = torch.linspace(1, 8, steps=8).view(4, 2)
print(x)
area1...# index=torch.max(area,1)[0]
b=x[area]
# b= x[torch.where((x[:,0] 0) & (x[:,0]<6))]
# print(b)
以上这篇使用...pytorch 筛选出一定范围的值就是小编分享给大家的全部内容了,希望能给大家一个参考。