Further debugging showed me that it’s not always the same column (but seems to be always only 1 column on layer 2) causing the “explosion” of nan.
Could you please help me understand what might be causing this and how to fix?