KeyValueSort

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>
create(Scope scope, Operand<T> keys, Operand<U> values)
Factory method to create a class wrapping a new KeyValueSort operation.
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

public Output<T> sortedKeys ()

A `Tensor` of type K.

public Output<U> sortedValues ()

A `Tensor` of type V.