nanmean

nanmean(data: Tensor, dim: tuple[int, ...] | int | None = None, keepdim: bool = False, **kwargs) → Tensor

Returns a tensor with dimensions reduced by taking the arithmetic mean, excluding NaN values.

This is a wrapper around torch.nanmean().

Parameters:

• data (torch.Tensor) – tensor to which operations should be applied.

• dim (tuple[int, ...] | int | None, optional) – dimension(s) along which the reduction should be applied, all dimensions when None. Defaults to None.

• keepdim (bool, optional) – if the dimensions should be retained in the output. Defaults to False.

Returns:

dimensionally reduced tensor.

Return type:

torch.Tensor