-
Notifications
You must be signed in to change notification settings - Fork 160
Open
Description
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
Labels
No labels