Description

Order that inTensor will be sorted when applying TensorMath.argSortMasked().

Example

var size = TensorMath.argSortMasked(inTensor, mask, outTensor, order); 
print(outTensor.subarray(0, size));
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]
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]

Enumeration Members

Enumeration Members

Ascending: number

Description

Applies ascending sorting order when returning indices in TensorMath.argSortMasked().

Descending: number

Generated using TypeDoc