I am also confused and on pytorch forums