nanmedian

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

Returns a tensor with dimensions reduced by taking median, excluding NaN values.

This is an alias for nanquantile(..., q=0.5, interpolation="midpoint").

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