abs2

dnois.torch.abs2(x: Tensor) → Tensor

Computes the squared magnitude of a complex tensor.

Parameters:

x (Tensor) – A complex-valued tensor.

Returns:

Squared magnitude.

Return type:

Tensor