What’s more, this is my network structure:
import torch
from torch.utils.data import Dataset, DataLoader
import numpy as np
import torch.nn as nn
import torch.nn.functional as F

class fNet(nn.Module):
    def __init__(self):
        super(fNet,self).__init__()
        self.fc0=nn.Linear(1,300)

        self.fc1_i=nn.Linear(1,300,bias=False)
        self.fc1=nn.Linear(300,300)

        self.fc2_i=nn.Linear(1,1,bias=False)
        self.fc2=nn.Linear(300,1)
    def forward(self,x):
        """
         Args:
             self: the Net's layers
             x:x0
        """
        x0=x.view(-1,1)
        
        x1=F.leaky_relu(self.fc0(x0))
        x1=F.leaky_relu(self.fc1(x1)+self.fc1_i(x0))
        x1=self.fc2(x1)+self.fc2_i(x0)
        return x1