Ejemplos de Tensor.mean en Python

Ejemplo n.º 1

def test_arctanh(t: Tensor) -> Tensor:
    return ep.arctanh((t - t.mean()) / t.max())

Ejemplo n.º 2

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