I was implementing a new idea. I want to implement a linear layer which takes the 3-dimensional RGB image and multiplies a 3-dimensional linear layer but with one to one cel