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来说,它有以下三种使用情况: