Rate this Page
★★★★★
torch.msort#
- torch.msort(input:Tensor,*,out:Optional[Tensor])→Tensor#
Sorts the elements of the
inputtensor along its first dimensionin ascending order by value.Note
torch.msort(t) is equivalent totorch.sort(t, dim=0)[0].See also
torch.sort().- Parameters
input (Tensor) – the input tensor.
- Keyword Arguments
out (Tensor,optional) – the output tensor.
Example:
>>>t=torch.randn(3,4)>>>ttensor([[-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]])
On this page