Okay. Then we can modify the function according to our need and save the file in torch.nn. After that we can call the function ‘LinearFunction’ in ‘Linear’ with the help of ‘apply’ (LinearFunction.apply). am I correct ?