从Pytorch张量获取值的范围

我试图从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)

---

new_tensor2=tensor[tensor2和tensor=-3和x=-3和x=-3和x您可以使用
和
操作

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

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)]))

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.])