Rate this Page

torch.msort#

torch.msort(input:Tensor,*,out:Optional[Tensor])Tensor#

Sorts the elements of theinput tensor along its first dimensionin ascending order by value.

Note

torch.msort(t) is equivalent totorch.sort(t, dim=0)[0].See alsotorch.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]])