Pytorch

shape