def test_arctanh(t: Tensor) -> Tensor: return ep.arctanh((t - t.mean()) / t.max())
def mean(self, tag: str, x: ep.Tensor, step: int) -> None: self.writer.add_scalar(tag, x.mean(axis=0).item(), step)