complexPyTorch icon indicating copy to clipboard operation
complexPyTorch copied to clipboard

Complex MSELoss

Open dstark1993 opened this issue 2 years ago • 0 comments

Similar to torch.nn.MSELoss().

I guess the function is pretty obvious as seen in https://github.com/pytorch/pytorch/issues/46642

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

dstark1993 avatar Feb 01 '23 10:02 dstark1993