tensor中的比較

比較函數中有一些是逐元素比較，操作類似于逐元素操作，還有一些則類似于歸并操作。常用比較函數如表3-6所示。

常用比較函數

函數功能

gt/lt/ge/le/eq/ne	大于/小于/大于等于/小于等于/等于/不等
topk	最大的k個數
sort	排序
max/min	比較兩個tensor最大最小值

表中第一行的比較操作已經實現了運算符重載，因此可以使用a>=b、a>b、a!=b、a==b，其返回結果是一個ByteTensor，可用來選取元素。max/min這兩個操作比較特殊，以max來說，它有以下三種使用情況：

• t.max(tensor)：返回tensor中最大的一個數
• t.max(tensor,dim)：指定維上最大的數，返回tensor和下標
• t.max(tensor1, tensor2): 比較兩個tensor相比較大的元素

至于比較一個tensor和一個數，可以使用clamp函數。下面舉

總結

以上是生活随笔為你收集整理的pytorch Tensor的操作和Numpy之间的转化（三）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。