Pytorch Weighted Stuffed