pytorch Tensor的操作和Numpy之间的转化(三)
生活随笔
收集整理的這篇文章主要介紹了
pytorch Tensor的操作和Numpy之间的转化(三)
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
tensor中的比較
比較函數(shù)中有一些是逐元素比較,操作類似于逐元素操作,還有一些則類似于歸并操作。常用比較函數(shù)如表3-6所示。
常用比較函數(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)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 厦门银行是正规银行吗 在哪一年上市的
- 下一篇: cmd 命令(上)