Hi All,
I have a layer for the U-Net model; the kernel of the layer must loop through all the pixels in the image (image size is 1024x1024) -  it is really slow. It takes 2.15 (minutes on average) for one image per one iteration. Any suggestions on how can I optimize the double loop over the image matrix?