在没有for循环的情况下，如何将自定义2D层传递到Keras中3D张量的每个通道？

Question

假设我在Keras中有一个自定义2D层，该层采用(?, n_h, n_w)形状的张量，并创建(?, m_h, m_w)形状的张量

如何将形状为(?, n_h, n_w, n_c)的三维张量的每个通道通过该层，以获得形状为(?, m_h, m_w, n_c)的三维张量

请注意，自定义层将使用不同的权重集称为n_c time

另外，请注意，我不希望使用for循环一次一个循环地通过每个通道