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

Question

有没有办法将不同的密集层应用于输入的每个通道？
即输入张量为 ，[batch,height,width,channels]我想对每个通道应用不同的密集层，从而具有#channels各层，每个层的输入为height*width。
从张量流密集文档中我看到

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

在这种情况下，这不是期望的结果。