使用pytorch 筛选出一定范围的值

时间:2022-07-27
本文章向大家介绍使用pytorch 筛选出一定范围的值，主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项，具有一定的参考价值，需要的朋友可以参考一下。

我就废话不多说了，大家还是直接看代码吧~

import torch
input_tensor = torch.tensor([1,2,3,4,5])
print(input_tensor 3)
mask = (input_tensor 3).nonzero()
print(mask)
print(input_tensor.index_select(0,mask))
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=(x[:,0] 5.5)&(x[:,1] 5.5)

c=x[:,0]*x[:,1]
area2=c 25

area=area1|area2
print(x[area])

if 0:
# index=torch.max(area,1)[0]
b=x[area]
# b= x[torch.where((x[:,0] 0) & (x[:,0]<6))]
# print(b)

以上这篇使用pytorch 筛选出一定范围的值就是小编分享给大家的全部内容了，希望能给大家一个参考。

随机文章
本站知识点必读
最近更新