Enumeration SortOrder

Example

var size = TensorMath.argSortMasked(inTensor, mask, outTensor, order); 
print(outTensor.subarray(0, size)); 
Copy

var inTensor = [5.0, 2.0, 1.0, 3.0, 7.0, -1.0, -5.0]; 
var mask = [1.0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0]; 
var outTensor = new Uint32Array(7); 
var order = SortOrder.Ascending; 
var size = TensorMath.argSortMasked(inTensor, mask, outTensor, order);

print(outTensor.subarray(0, size)); // Expected: [2, 1, 3, 0, 4]
Copy

var inTensor = [5.0, 2.0, 1.0, 3.0, 7.0, -1.0, -5.0]; 
var mask = [1.0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0]; 
var outTensor = new Uint32Array(7); 
var order = SortOrder.Descending; 
var size = TensorMath.argSortMasked(inTensor, mask, outTensor, order); 

print(outTensor.subarray(0, size)); // Expected: [4, 0, 3, 1, 2]
Copy