Skip to content

Complex MSELoss #27

@dstark1993

Description

@dstark1993

Similar to torch.nn.MSELoss().

I guess the function is pretty obvious as seen in pytorch/pytorch#46642

def complex_mse_loss(output, target):
    return (0.5*(output - target)**2).mean(dtype=torch.complex64)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions