Expansion to Multi channel input

[wolke26]

Hi, I have a very basic question: Is your model capable of multi channel inputs? (RBG,etc..) I would be interested in checking out multi channel learning with your network. In your example you defined a "BayesianFlowNetwork_2D()" Version and in your model code its purely "BayesianFlowNetwork()". Thanks in advance and for sharing your work!