我训练了一个神经网络，然后想把这个神经网络保存成函数或者建立一个函数包含了这个神经网络，该怎么建这个函数？有没有大神指点一下，万分感谢🙏