block¶

Module Contents¶

Simple MLP block with residual connection.

_get_activation_fn(activation)

_get_activation_fn(activation)¶

class ResBlock(in_features, out_features, activation='relu')¶

Bases: torch.nn.Module

Simple MLP block with residual connection.

Parameters:

in_features – the feature dimension of each output sample.
out_features – the feature dimension of each output sample.
activation – the activation function of intermediate layer, relu or gelu.

in_features¶

out_features¶

fc1¶

fc2¶

bn¶

activation¶

forward(x)¶