Method from_tensor

backends/test/harness/error_statistics.py:30–40

Creates a TensorStatistics object from a tensor.

(cls, tensor: torch.Tensor)

    @classmethod
    def from_tensor(cls, tensor: torch.Tensor) -> "TensorStatistics":
        """Creates a TensorStatistics object from a tensor."""
        flattened = torch.flatten(tensor)
        return cls(
            shape=tensor.shape,
            numel=tensor.numel(),
            median=torch.median(flattened).item(),
            mean=flattened.mean().item(),
            max=flattened.max().item(),
            min=flattened.min().item(),
        )


@dataclass