Hello everyone,
In the call method of the custom layer, I have shape of the input into which first dimension is batch. I am dividing the images into the 'g' groups to get subfeatures.
Let's say image is (1, 7, 7, 1024) and g=4. So, shape will be =(1,7,7,4,256). Now, I want to use these shape as (1*4,7,7,256). This causes the graph execution earror with baseline model as batches are nonetype. Further, it works completely with any shape image with 4 dimensions.
I have checked few solutions in the stack overflow. But, not working as batch shape is NoneType and g is integer.
Thank you, any help will be appreciated.