Table Of Contents
Table Of Contents

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 fn recursively 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 ParameterDict containing this Block and all of its children’s Parameters(default), also can returns the select ParameterDict which 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 HybridBlock s 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 Parameter s of this Block and 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 Block and 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

name

Name of this Block, without ‘_’ in the end.

params

Returns this Block’s parameter dictionary (does not include its children’s parameters).

prefix

Prefix of this Block.