Changed torch.maximum to np.maximum

    import torch

    def loss(output, labels):

      import numpy as np

      output, labels = _make_pytorch_shapes_consistent(output, labels)

      return torch.mean(

          torch.pow(

              torch.maximum(1 - torch.multiply(labels, output),

                            torch.tensor(0)), 2),

              np.maximum(1 - torch.multiply(labels, output), torch.tensor(0)),

              2),

          dim=-1)

    return loss