Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge

Bucketize

public final class Bucketize

Bucketizes 'input' based on 'boundaries'.

For example, if the inputs are boundaries = [0, 10, 100] input = [[-5, 10000] [150, 10] [5, 100]]

then the output will be output = [[0, 3] [3, 2] [1, 3]]

Public Methods

Output<Integer>
asOutput()
Returns the symbolic handle of a tensor.
static <T extends Number> Bucketize
create(Scope scope, Operand<T> input, List<Float> boundaries)
Factory method to create a class wrapping a new Bucketize operation.
Output<Integer>
output()
Same shape with 'input', each value of input replaced with bucket index.

Inherited Methods

Public Methods

public Output<Integer> 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 Bucketize create (Scope scope, Operand<T> input, List<Float> boundaries)

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

Parameters
scope current scope
input Any shape of Tensor contains with int or float type.
boundaries A sorted list of floats gives the boundary of the buckets.
Returns
  • a new instance of Bucketize

public Output<Integer> output ()

Same shape with 'input', each value of input replaced with bucket index.