Pytorch summary.