@Operator public final class Bucketize extends PrimitiveOp implements Operand<Integer>
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]]
operation
Modifier and Type | Method and Description |
---|---|
Output<Integer> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T extends Number> |
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.
|
equals, hashCode, op, toString
public static <T extends Number> Bucketize create(Scope scope, Operand<T> input, List<Float> boundaries)
scope
- current scopeinput
- Any shape of Tensor contains with int or float type.boundaries
- A sorted list of floats gives the boundary of the buckets.public Output<Integer> output()
public Output<Integer> asOutput()
Operand
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.
asOutput
in interface Operand<Integer>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.