About video memory of get_correlation_layer in flownet.py

[DianaLi96]

Hello, here I met such issues: I use your implementation in my own code ,my displacement value set 20, and stride set 1,I will get OOM error, if I decrease the two values, the OOM error disappears. May I ask if there is another implementation to avoid the video memory overflow issues.