[bug-fix] for batch_size=1, squeeze() operation also squeezes the fir…#100

Open

timbektu wants to merge 1 commit intoomertov:mainfrom

timbektu:timbektu-patch-1

Commits on Jul 15, 2023

[bug-fix] for batch_size=1, squeeze() operation also squeezes the first (batch_size) dimension, and the code breaks when using batch_size =1. patched the bug in this commit.
timbektu
authored