5.tensor的排序与取极值

代码如下：

import torch

if __name__ == '__main__':

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

    #1.排序sort()函数：
    b=a.sort(0,True)#0表示按行排序，即对每一列的元素进行排序，True表示降序，False表示升序
    print("b:")
    print(b[0])
    print(b[1])

    #2.最值max()函数和min()函数
    c=a.max(0)##0表示按行选取最大值，即对每一列的元素取最大值
    print(c)

    d=a.min(1)#1表示按列选取最小值，即对每一行的元素取最小值
    print(d)