从Pytorch张量获取值的范围

Question

我正在尝试从pytorch张量中获取特定范围的值。

tensor=torch.tensor([0,1,2,3,4,5,6,7,8,9])
new_tensor=tensor[tensor>2]
print(new_tensor)

这将给出一个标量为3-9的张量

new_tensor2=tensor[tensor<8]
print(new_tensor2)

这将给出一个标量为0-7的张量

new_tensor3=tensor[tensor>2 and tensor<8]
print(new_tensor3)

但是，这会引发一个错误。我能用这样的东西得到一个值为3-7的张量吗？我正在尝试直接编辑张量，而不希望改变张量本身的顺序。

grad[x<-3]=0.1
grad[x>2]=1
grad[(x>=-3 and x<=2)]=siglrelu(grad[(x>=-3 and x<=2)])*(1.0-siglrelu(grad[(x>=-3 and x<=2)]))

这才是我真正要做的，我不是很确定该怎么做。感谢您的帮助，谢谢！

Answer 1

你可以使用&操作，

t = torch.arange(0., 10)
print(t)
print(t[(t > 2) & (t < 8)])

输出结果是，

tensor([0., 1., 2., 3., 4., 5., 6., 7., 8., 9.])
tensor([3., 4., 5., 6., 7.])