ValueError: cannot reshape array of size while aligning memory

sorry, that’s not the solution, I found out it’s just my personal mistake forgot to return tensor…