当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch msort用法及代码示例


本文简要介绍python语言中 torch.msort 的用法。

用法:

torch.msort(input, *, out=None) → Tensor

参数

input(Tensor) -输入张量。

关键字参数

out(Tensor,可选的) -输出张量。

input 张量的元素沿其第一个维度按值升序排序。

注意

torch.msort(t) 等价于 torch.sort(t, dim=0)[0] 。另见 torch.sort()

例子:

>>> t = torch.randn(3, 4)
>>> t
tensor([[-0.1321,  0.4370, -1.2631, -1.1289],
        [-2.0527, -1.1250,  0.2275,  0.3077],
        [-0.0881, -0.1259, -0.5495,  1.0284]])
>>> torch.msort(t)
tensor([[-2.0527, -1.1250, -1.2631, -1.1289],
        [-0.1321, -0.1259, -0.5495,  0.3077],
        [-0.0881,  0.4370,  0.2275,  1.0284]])

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.msort。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。