二维张量第0维排序是根据什么排的?
来源：3-11 比较运算-排序-topk-kthvalue-数据合法性校验
慕工程3141378
2021-05-12
rand = torch.tensor([[0.0466, 0.0847], [0.8495, 0.0723]])
sorted_rand1 = torch.sort(rand, 1)
#第一维排序好理解 就是第2个维度进行排序
#tensor([[0.0466, 0.0847],
# [0.0723, 0.8495]])
sorted_rand = torch.sort(rand, 0)
#为什么结果是如下啊 不理解第0维排序是下面的结果,根据什么规则排的啊?
#tensor([[0.0466, 0.0723],
# [0.8495, 0.0847]])
写回答
1回答
-
可以以矩形为例子去理解，不同的纬度是指矩形的行和列，取不同的纬度做排序，就是，按照行或者列排序，具体操作就是取每一行的数据进行排序，或者取每一列的数据进行排序。
012021-05-12
相似问题