def hook(self, m:nn.Module, i:Tensors, o:Tensors)->Tuple[Rank0Tensor,Rank0Tensor]:
        "Take the mean and std of `o`."
        return o.mean().item(),o.std().item()
def hook ( self , m : nn . Module , i : Tensors , o : Tensors ) -> Tuple [ Rank0Tensor , Rank0Tensor ] : return o . mean ( ) . item ( ) , o . std ( ) . item ( )
Take the mean and std of `o`.