[0a9449]: / docs / source / notes / encoders / mlp.rst

Download this file

35 lines (16 with data), 813 Bytes

MLP

class DeepPurpose.models.MLP(nn.Sequential)

Multi-Layer Perceptron (MLP) (Source) is a class of feedforward artificial neural network. An MLP consists of at least three layers of nodes: an input layer, a hidden layer and an output layer.

constructor create an instance of MLP

__init__(self, input_dim, hidden_dim, hidden_dims)
  • input_dim (int) - dimension of input feature.
  • hidden_dim (int) - dimension of hidden layer.

Calling functions implement the feedforward procedure of MLP.

forward(self, v)
  • v (torch.Tensor) - input feature of MLP.