Store And Forward Method Pytorch