mxnet.gluon.rnn.DropoutCell¶
-
class
mxnet.gluon.rnn.DropoutCell(rate, axes=(), prefix=None, params=None)[source]¶ Applies dropout on input.
- Parameters
rate (float) – Percentage of elements to drop out, which is 1 - percentage to retain.
axes (tuple of int, default ()) – The axes on which dropout mask is shared. If empty, regular dropout is applied.
- Inputs:
data: input tensor with shape (batch_size, size).
states: a list of recurrent state tensors.
- Outputs:
out: output tensor with shape (batch_size, size).
next_states: returns input states directly.
-
__init__(rate, axes=(), prefix=None, params=None)[source]¶ Initialize self. See help(type(self)) for accurate signature.
Methods
__init__(rate[, axes, prefix, params])Initialize self.
apply(fn)Applies
fnrecursively to every child block as well as self.begin_state([batch_size, func])Initial state for this cell.
cast(dtype)Cast this Block to use another data type.
collect_params([select])Returns a
ParameterDictcontaining thisBlockand all of its children’s Parameters(default), also can returns the selectParameterDictwhich match some given regular expressions.export(path[, epoch])Export HybridBlock to json format that can be loaded by SymbolBlock.imports, mxnet.mod.Module or the C++ interface.
forward(inputs, states)Unrolls the recurrent cell for one time step.
hybrid_forward(F, inputs, states)Overrides to construct symbolic graph for this Block.
hybridize([active])Activates or deactivates
HybridBlocks recursively.infer_shape(*args)Infers shape of Parameters from inputs.
infer_type(*args)Infers data type of Parameters from inputs.
initialize([init, ctx, verbose, force_reinit])Initializes
Parameters of thisBlockand its children.load_parameters(filename[, ctx, …])Load parameters from file previously saved by save_parameters.
load_params(filename[, ctx, allow_missing, …])[Deprecated] Please use load_parameters.
name_scope()Returns a name space object managing a child
Blockand parameter names.register_child(block[, name])Registers block as a child of self.
register_forward_hook(hook)Registers a forward hook on the block.
register_forward_pre_hook(hook)Registers a forward pre-hook on the block.
reset()Reset before re-using the cell for another graph.
save_parameters(filename)Save parameters to file.
save_params(filename)[Deprecated] Please use save_parameters.
state_info([batch_size])shape and layout information of states
summary(*inputs)Print the summary of the model’s output and parameters.
unroll(length, inputs[, begin_state, …])Unrolls an RNN cell across time steps.
Attributes
nameName of this
Block, without ‘_’ in the end.paramsReturns this
Block’s parameter dictionary (does not include its children’s parameters).prefixPrefix of this
Block.