def forward(self, input_tensor):
    'Forward pass on the neural network model.\n\n        Args:\n            input_tensor: the input tensor.\n\n        Returns:\n            Tensor with model results.\n        '
    output = F.relu(self.input_layer(input_tensor))
    output = self.hidden_layers(output)
    output = self.output_layer(output)
    return output
Forward pass on the neural network model.

Args:
    input_tensor: the input tensor.

Returns:
    Tensor with model results.
def forward(self, input_tensor):
    'Forward pass on the neural network model.\n\n        Args:\n            input_tensor: the input tensor.\n\n        Returns:\n            Tensor with model results.\n        '
    output = F.relu(self.input_layer(input_tensor))
    output = self.hidden_layers(output)
    output = self.output_layer(output)
    return output