张量流中每个通道的完全连接层

时间:2019-03-26 07:48:02

标签: python tensorflow

是否可以对输入的每个通道应用不同密集层?
即输入张量为[batch,height,width,channels]的情况下,我想向每个通道应用不同的密集层,因此要拥有#channels层,每个层的输入为height*width
从tensorflow密集文档中,我看到

Note: that if inputs have a rank greater than 2, then inputs is flattened prior to the initial matrix multiply by weights.

在这种情况下这不是理想的结果。

0 个答案:

没有答案