StatelessMultinomial

Stay organized with collections Save and categorize content based on your preferences.
public final class StatelessMultinomial

Draws samples from a multinomial distribution.

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

Output<V>
asOutput()
Returns the symbolic handle of the tensor.
static StatelessMultinomial<TInt64>
create(Scope scope, Operand<? extends TNumber> logits, Operand<TInt32> numSamples, Operand<? extends TNumber> seed)
Factory method to create a class wrapping a new StatelessMultinomial operation using default output types.
static <V extends TNumber> StatelessMultinomial<V>
create(Scope scope, Operand<? extends TNumber> logits, Operand<TInt32> numSamples, Operand<? extends TNumber> seed, Class<V> outputDtype)
Factory method to create a class wrapping a new StatelessMultinomial operation.
Output<V>
output()
2-D Tensor with shape `[batch_size, num_samples]`.

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "StatelessMultinomial"

Public Methods

public Output<V> asOutput ()

Returns the symbolic handle of the 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 StatelessMultinomial<TInt64> create (Scope scope, Operand<? extends TNumber> logits, Operand<TInt32> numSamples, Operand<? extends TNumber> seed)

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

Parameters
scope current scope
logits 2-D Tensor with shape `[batch_size, num_classes]`. Each slice `[i, :]` represents the unnormalized log probabilities for all classes.
numSamples 0-D. Number of independent samples to draw for each row slice.
seed 2 seeds (shape [2]).
Returns
  • a new instance of StatelessMultinomial

public static StatelessMultinomial<V> create (Scope scope, Operand<? extends TNumber> logits, Operand<TInt32> numSamples, Operand<? extends TNumber> seed, Class<V> outputDtype)

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

Parameters
scope current scope
logits 2-D Tensor with shape `[batch_size, num_classes]`. Each slice `[i, :]` represents the unnormalized log probabilities for all classes.
numSamples 0-D. Number of independent samples to draw for each row slice.
seed 2 seeds (shape [2]).
Returns
  • a new instance of StatelessMultinomial

public Output<V> output ()

2-D Tensor with shape `[batch_size, num_samples]`. Each slice `[i, :]` contains the drawn class labels with range `[0, num_classes)`.