我有个问题，例子是这样：一个正方体的上表面被分割成128x128的离散网格，每个网格里有随机载荷，输出是每个网格在随机载荷下的形变位移量，这样的正方体有1000个样本。简单说就是，输入是128x128的载荷矩阵，输出也是128x128的位移矩阵，这样的样本数1000，对其进行训练，实现输入的载荷矩阵预测输出的形变位移矩阵。我想的是使用卷积神经网络，请问这个怎实现了？或者有其他方法吗？谢谢
输出调整成（1,128*128），或者输出分块，分别输出