org.tensorflow.op.nn

Class CudnnRnnParamsToCanonical<T extends Number>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.nn.CudnnRnnParamsToCanonical<T>

Type Parameters:

T - data type for weights() output

All Implemented Interfaces:

Op

@Operator(group="nn")
public final class CudnnRnnParamsToCanonical<T extends Number>
extends PrimitiveOp

Retrieves CudnnRNN params in canonical form.

Retrieves a set of weights from the opaque params buffer that can be saved and restored in a way compatible with future runs.

Note that the params buffer may not be compatible across different GPUs. So any save and restoration should be converted to and from the canonical weights and biases.

num_layers: Specifies the number of layers in the RNN model. num_units: Specifies the size of the hidden state. input_size: Specifies the size of the input state. num_params: number of parameter sets for all layers. Each layer may contain multiple parameter sets, with each set consisting of a weight matrix and a bias vector. weights: the canonical form of weights that can be used for saving and restoration. They are more likely to be compatible across different generations. biases: the canonical form of biases that can be used for saving and restoration. They are more likely to be compatible across different generations. rnn_mode: Indicates the type of the RNN model. input_mode: Indicate whether there is a linear projection between the input and The actual computation before the first layer. 'skip_input' is only allowed when input_size == num_units; 'auto_select' implies 'skip_input' when input_size == num_units; otherwise, it implies 'linear_input'. direction: Indicates whether a bidirectional model will be used. dir = (direction == bidirectional) ? 2 : 1 dropout: dropout probability. When set to 0., dropout is disabled. seed: the 1st part of a seed to initialize dropout. seed2: the 2nd part of a seed to initialize dropout.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	CudnnRnnParamsToCanonical.Options Optional attributes for CudnnRnnParamsToCanonical

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
List<Output<T>>	biases()
static <T extends Number> CudnnRnnParamsToCanonical<T>	create(Scope scope, Operand<Integer> numLayers, Operand<Integer> numUnits, Operand<Integer> inputSize, Operand<T> params, Long numParams, CudnnRnnParamsToCanonical.Options... options) Factory method to create a class wrapping a new CudnnRnnParamsToCanonical operation.
static CudnnRnnParamsToCanonical.Options	direction(String direction)
static CudnnRnnParamsToCanonical.Options	dropout(Float dropout)
static CudnnRnnParamsToCanonical.Options	inputMode(String inputMode)
static CudnnRnnParamsToCanonical.Options	rnnMode(String rnnMode)
static CudnnRnnParamsToCanonical.Options	seed(Long seed)
static CudnnRnnParamsToCanonical.Options	seed2(Long seed2)
List<Output<T>>	weights()

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

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

Method Detail

create

public static <T extends Number> CudnnRnnParamsToCanonical<T> create(Scope scope,
                                                                     Operand<Integer> numLayers,
                                                                     Operand<Integer> numUnits,
                                                                     Operand<Integer> inputSize,
                                                                     Operand<T> params,
                                                                     Long numParams,
                                                                     CudnnRnnParamsToCanonical.Options... options)

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

Parameters:

scope - current scope

numLayers -

numUnits -

inputSize -

params -

numParams -

options - carries optional attributes values

Returns:

a new instance of CudnnRnnParamsToCanonical

rnnMode

public static CudnnRnnParamsToCanonical.Options rnnMode(String rnnMode)

Parameters:

rnnMode -

inputMode

public static CudnnRnnParamsToCanonical.Options inputMode(String inputMode)

Parameters:

inputMode -

direction

public static CudnnRnnParamsToCanonical.Options direction(String direction)

Parameters:

direction -

dropout

public static CudnnRnnParamsToCanonical.Options dropout(Float dropout)

Parameters:

dropout -

seed

public static CudnnRnnParamsToCanonical.Options seed(Long seed)

Parameters:

seed -

seed2

public static CudnnRnnParamsToCanonical.Options seed2(Long seed2)

Parameters:

seed2 -

weights

public List<Output<T>> weights()

biases

public List<Output<T>> biases()