rnnfc

CPUStaticRNN & its config.

class common.optim.ne.net.cpu.static.rnnfc.CPUStaticRNNFCConfig(input_size, hidden_size, output_size)

Bases: object

Config values for CPUStaticRNNFC.

Parameters:

• input_size (int) – Size of the input tensor.

• hidden_size (int) – Size of the RNN hidden state.

• output_size (int) – Size of the output tensor.

class common.optim.ne.net.cpu.static.rnnfc.CPUStaticRNNFC(config)

Bases: Module

CPU-running static architecture RNN w/ a final FC layer.

Parameters:

config (CPUStaticRNNFCConfig)

reset()

Resets the hidden state of the RNN.

Return type:

None

forward(x)

Forward pass.

Return type:

Tensor