public final class
KeyValueSort
Wraps the XLA Sort operator, documented at
https://www.tensorflow.org/performance/xla/operation_semantics#sort .
Sorts a tensor. Currently only sorts in ascending order are supported.
Constants
String | OP_NAME | The name of this op, as known by TensorFlow core engine |
Public Methods
static <T extends TNumber, U extends TType> KeyValueSort<T, U> | |
Output<T> |
sortedKeys()
A `Tensor` of type K.
|
Output<U> |
sortedValues()
A `Tensor` of type V.
|
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
Constant Value:
"XlaKeyValueSort"
Public Methods
public static KeyValueSort<T, U> create (Scope scope, Operand<T> keys, Operand<U> values)
Factory method to create a class wrapping a new KeyValueSort operation.
Parameters
scope | current scope |
---|---|
keys | A `Tensor` of type K. |
values | A `Tensor` of type V. |
Returns
- a new instance of KeyValueSort