比较重要的是排序函数sort(),选择沿着指定维度进行排序,返回排序后的Tensor及对应索引位置。max()与min()函数则是沿着指定维度选择最大与最小元素,返回改元素及对应的索引位置。
对于Tensor的单元素数学运算,如abs()、sqrt()、log()、pow()和三角函数等,都是逐元素操作(element-wise),输出的Tensor形状与原始Tensor形状一致
对于类似求和、求均值、求方差、求距离等需要多个元素完成的操作,往往需要沿着某个维度进行计算,在Tensor中属于归并操作,输出形状小于输入形状。与NumPy极为相似。
PyTorch基础:Tensor的排序与取极值
最新推荐文章于 2024-04-12 20:42:33 发布