Skip to content

Complex MSELoss #27

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