tensor中的比較

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

常用比較函數(shù)

函數(shù)功能

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

表中第一行的比較操作已經(jīng)實(shí)現(xiàn)了運(yùn)算符重載，因此可以使用a>=b、a>b、a!=b、a==b，其返回結(jié)果是一個(gè)ByteTensor，可用來選取元素。max/min這兩個(gè)操作比較特殊，以max來說，它有以下三種使用情況：

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

至于比較一個(gè)tensor和一個(gè)數(shù)，可以使用clamp函數(shù)。下面舉

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。