index_select和clamp

1、torch.clamp(input,min,max,out=None)-> Tensor

     将input中的元素限制在[min,max]范围内并返回一个Tensor

2、index_select()

 x = torch.randn(3, 4)
 print(x)

indices = torch.LongTensor([0, 2]) y = torch.index_select(x, 0, indices) print(y)

z = torch.index_select(x, 1, indices) print(z)

参考:https://blog.csdn.net/appleml/article/details/78630452 
x = torch.randn(3, 4)
print(x)

indices = torch.LongTensor([0, 2])
y = torch.index_select(x, 0, indices)
print(y)

z = torch.index_select(x, 1, indices)
print(z)

猜你喜欢

转载自www.cnblogs.com/573177885qq/p/9121972.html