使用火炬绘制神经网络函数

Question

我已经使用pytorch构建和训练了一个神经网络，该神经网络将用作函数的数值模拟。

现在将其视为3个变量的函数：

• t作为1维欧几里德空间中的时间；
• x作为3维欧几里德空间中的位置；
• v作为3维欧几里德空间中的速度。

我想用固定x的{​​{1}}和v绘制此函数。谁能帮我提出一个想法或一些代码？我有义务。