src/transformers/models/resnet/modeling_resnet.py/ResNetBasicLayer/forward
class ResNetBasicLayer:    def forward(self, hidden_state):
        residual = hidden_state
        hidden_state = self.layer(hidden_state)
        residual = self.shortcut(residual)
        hidden_state += residual
        hidden_state = self.activation(hidden_state)
        return hidden_state