两个卷积网络（千层面）的权重之间的距离

Question

我有一个权重为A的预训练网络，而另一个权重为B的可训练网络。我想训练第二个网络，但要限制权重A和B之间的距离，即通过添加一个目标用于最小化的函数，例如sum（（AB）^ 2）。知道如何在千层面做这件事吗？

谢谢。