-
Notifications
You must be signed in to change notification settings - Fork 0
sort
Okerew edited this page Aug 2, 2024
·
3 revisions
You can use sort like here for tensors
from okrolearn.okrolearn import Tensor
# Create a tensor
t = Tensor([3, 1, 4, 1, 5, 9, 2, 6])
# Sort the tensor
sorted_t, indices = t.sort()
print("Sorted tensor:", sorted_t)
print("Sorting indices:", indices)
# Sort in descending order
sorted_t_desc, indices_desc = t.sort(descending=True)
print("Sorted tensor (descending):", sorted_t_desc)
print("Sorting indices (descending):", indices_desc)