如何在Swift中为Tensorflow创建卷积滤波器,其中只有矩阵的一部分是可训练的?

如何在Swift中为Tensorflow创建卷积滤波器,其中只有矩阵的一部分是可训练的?,swift,tensorflow,Swift,Tensorflow,本质上,我试图制作一个卷积滤波器,其中只有一部分是可训练的,如下所示: 0T0 000 T0T 其中0权重设置为零,T权重可训练。我可以将整个层设置为可训练的或不可训练的,但如何在卷积层中将单个权重设置为可训练的还不清楚 [编辑]。。。好的,我想我可能有一个解决办法 0T0 1,1 1,2 1,3 1,4 000 applied stride 1,1 to an image 2,1 2,2 2,3 2,4 T0T

本质上,我试图制作一个卷积滤波器,其中只有一部分是可训练的,如下所示:

0T0
000
T0T
其中0权重设置为零,T权重可训练。我可以将整个层设置为可训练的或不可训练的,但如何在卷积层中将单个权重设置为可训练的还不清楚

[编辑]。。。好的,我想我可能有一个解决办法

0T0                                1,1 1,2 1,3 1,4
000 applied stride 1,1 to an image 2,1 2,2 2,3 2,4 
T0T                                3,1 3,2 3,3 3,4
                                   4,1 4,2 4,3 4,4
相当于:

                                                 3,1 1,2 3,3 3,2 1,3 3,4
TTT applied stride 1,3 to the transformed image: 4,1 2,2 4,3 4,2 2,3 4,4 

换句话说,我可以变换输入张量,然后对变换后的输入张量执行正常卷积