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

Before Change

        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):
    Alias for :func:`scalar_mult`.

After Change

    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.

Instances