Tune in to the first Women in ML Symposium this Tuesday, October 19 at 9am PST Register now

UpperBound

public final class UpperBound

Applies upper_bound(sorted_search_values, values) along each row.

Each set of rows with the same index in (sorted_inputs, values) is treated independently. The resulting row is the equivalent of calling `np.searchsorted(sorted_inputs, values, side='right')`.

The result is not a global index to the entire `Tensor`, but rather just the index in the last dimension.

A 2-D example: sorted_sequence = [[0, 3, 9, 9, 10], [1, 2, 3, 4, 5]] values = [[2, 4, 9], [0, 2, 6]]

result = UpperBound(sorted_sequence, values)

result == [[1, 2, 4], [0, 2, 5]]

Public Methods

Output<U>
asOutput()
Returns the symbolic handle of a tensor.
static <U extends Number, T> UpperBound<U>
create(Scope scope, Operand<T> sortedInputs, Operand<T> values, Class<U> outType)
Factory method to create a class wrapping a new UpperBound operation.
static <T> UpperBound<Integer>
create(Scope scope, Operand<T> sortedInputs, Operand<T> values)
Factory method to create a class wrapping a new UpperBound operation using default output types.
Output<U>
output()
A `Tensor` with the same shape as `values`.

Inherited Methods

Public Methods

public Output<U> asOutput ()

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

public static UpperBound<U> create (Scope scope, Operand<T> sortedInputs, Operand<T> values, Class<U> outType)

Factory method to create a class wrapping a new UpperBound operation.

Parameters
scope current scope
sortedInputs 2-D Tensor where each row is ordered.
values 2-D Tensor with the same numbers of rows as `sorted_search_values`. Contains the values that will be searched for in `sorted_search_values`.
Returns
  • a new instance of UpperBound

public static UpperBound<Integer> create (Scope scope, Operand<T> sortedInputs, Operand<T> values)

Factory method to create a class wrapping a new UpperBound operation using default output types.

Parameters
scope current scope
sortedInputs 2-D Tensor where each row is ordered.
values 2-D Tensor with the same numbers of rows as `sorted_search_values`. Contains the values that will be searched for in `sorted_search_values`.
Returns
  • a new instance of UpperBound

public Output<U> output ()

A `Tensor` with the same shape as `values`. It contains the last scalar index into the last dimension where values can be inserted without changing the ordered property.