db073909d05ae6f91b7abd8d6251bfa247349447,qucumber/utils/cplx.py,,conjugate,#Any#,226

Before Change

        z[0] = x[0]
        z[1] = -x[1]

    if x.dim() == 3:
        z = torch.zeros(2, x.size()[2], x.size()[1], dtype=x.dtype, device=x.device)
        z[0] = torch.transpose(x[0], 0, 1)
        z[1] = -torch.transpose(x[1], 0, 1)

    return z


def elementwise_mult(x, y):

After Change

    :returns: The conjugate of x.
    :rtype: torch.Tensor
    
    if x.dim() < 3:
        return conj(x)
    else:
        return make_complex(
            torch.transpose(real(x), 0, 1), -torch.transpose(imag(x), 0, 1)
        )


def conj(x):
    Returns the element-wise complex conjugate of the argument.

    :param x: A complex tensor.

Instances