Function complex_abs_t

monai/apps/reconstruction/complex_utils.py:102–114

Compute the absolute value of a complex tensor. Args: x: Input tensor with 2 channels in the last dimension representing real and imaginary parts. Returns: Absolute value along the last dimension

(x: Tensor)

def complex_abs_t(x: Tensor) -> Tensor:
    """
    Compute the absolute value of a complex tensor.

    Args:
        x: Input tensor with 2 channels in the last dimension representing real and imaginary parts.

    Returns:
        Absolute value along the last dimension
    """
    if x.shape[-1] != 2:
        raise ValueError(f"x.shape[-1] is not 2 ({x.shape[-1]}).")
    return (x[..., 0] ** 2 + x[..., 1] ** 2) ** 0.5  # type: ignore


def complex_abs(x: NdarrayOrTensor) -> NdarrayOrTensor: