says:

TypeError: dstack() takes 1 positional argument but 2 were given

EDIT: Btw, it seems the second suggestion you have given worked out!
EDIT2:Actually nvm, im getting loss Nans for the second suggestion