Have a question? Connect with the community at the TensorFlow Forum Visit Forum

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.