org.tensorflow.op.random

Class StatelessMultinomial<V extends TNumber>

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.random.StatelessMultinomial<V>

Type Parameters:

V - data type for output output

All Implemented Interfaces:

org.tensorflow.ndarray.Shaped, Op, Operand<V>

@Operator(group="random")
public final class StatelessMultinomial<V extends TNumber>
extends RawOp
implements Operand<V>

Draws samples from a multinomial distribution.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	StatelessMultinomial.Inputs

Field Summary

Fields

Modifier and Type	Field and Description
static String	OP_NAME The name of this op, as known by TensorFlow core engine

Fields inherited from class org.tensorflow.op.RawOp

operation

Constructor Summary

Constructors

Constructor and Description
StatelessMultinomial(Operation operation)

Method Summary

Modifier and Type	Method and Description
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, with the 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() Gets output.

Methods inherited from class org.tensorflow.op.RawOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.tensorflow.Operand

asTensor, shape, type

Methods inherited from interface org.tensorflow.op.Op

env, op

Methods inherited from interface org.tensorflow.ndarray.Shaped

rank, size

Field Detail

OP_NAME

public static final String OP_NAME

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

See Also:

Constant Field Values

Constructor Detail

StatelessMultinomial

public StatelessMultinomial(Operation operation)

Method Detail

create

@Endpoint(describeByClass=true)
public 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.

Type Parameters:

V - data type for StatelessMultinomial output and operands

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]).

outputDtype - The value of the outputDtype attribute

Returns:

a new instance of StatelessMultinomial

create

@Endpoint(describeByClass=true)
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, with the 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, with default output types

output

public Output<V> output()

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

Returns:

output.

asOutput

public Output<V> asOutput()

Description copied from interface: Operand

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.

Specified by:

asOutput in interface Operand<V extends TNumber>

See Also:

OperationBuilder.addInput(Output)