Restore data format from SCB to SSCB after patch embedding

I'm trying to create a neural network with the Deep Network Designer app but I'm having issues with data dimensions through the net itself.

Since the self attention layer needs SCB format data as input, I'm using a patch embedding layer to do so. The problem is that I want to restore the original data format, i.e. SSCB, after the normalization layer, but I'm not able to find a way to do so.

Thank you in advance for your help.

Answer 1

[Matt J]

One way is to use a functionLayer.