MLP

class vformer.decoder.mlp.MLPDecoder(config=(1024,), n_classes=10)[source]

Bases: Module

Parameters

  • config (int or tuple or list) – Configuration of the hidden layer(s)

  • n_classes (int) – Number of classes for classification

forward(x)[source]
Parameters

x (torch.Tensor) – Input tensor

Returns

Returns output tensor of size n_classes, Note that torch.nn.Softmax is not applied to the output tensor.

Return type

torch.Tensor