Order that inTensor will be sorted when applying TensorMath.argSortMasked().
TensorMath.argSortMasked()
var size = TensorMath.argSortMasked(inTensor, mask, outTensor, order); print(outTensor.subarray(0, size)); Copy
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] 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]
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
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]
Applies ascending sorting order when returning indices in TensorMath.argSortMasked().
Order that inTensor will be sorted when applying
TensorMath.argSortMasked()
.Example