Revert "Remove Torch Summary"