我正在尝试使用Keras功能API实现自己的层。想法是转置卷积运算符。到目前为止,我得到了:
k1 = self.tied_to.kernel
self.kernel = K.transpose(k1)
如果我仅传递前一层的内核而不对其进行任何处理,则该代码有效。内核以3x3形状定义,因此对其进行转置应该很好。但是问题在于内核是具有3x3x1x1形状的张量的形式,并且K.transpose
转置了整个张量。如何只转置内核本身,以便再次获得3x3x1x1形式的内核?
您要使用K.permute_dimensions
。
我不确定您要转置的内容,但我看到两种可能性。
K.permute_dimensions(k1, (1,0,2,3))
K.permute_dimensions(k1, (0,1,3,2))
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句