Edit:  Current question in my last post down below!
Is this function correct or does it apply
the relu to the input layer and is therefore wrong?

def forward(self, input):
x = self.fc1(input)
x = nn.functional.relu(x)
x = self.fc2(x)
x = nn.functional.relu(x)
x = self.fc3(x)
output = nn.functional.relu(x)
return output

Sry if this is too basic but I’m still new to machine learning.
An advice would be much appreciated. Gomen!