MNISTMLP PNN#

Classes#

class backbone.MNISTMLP_PNN.MNISTMLP_PNN(input_size, output_size, old_cols=None)[source]#

Bases: MammothBackbone

Network composed of two hidden layers, each containing 100 ReLU activations. Designed for the MNIST dataset, equipped with lateral connection.

forward(x, returnt='out')[source]#

Compute a forward pass.

Parameters:

x (Tensor) – input tensor (batch_size, input_size)

Return type:

Tensor

Retruns:

output tensor (output_size)

reset_parameters()[source]#

Calls the Xavier parameter initialization function.