returnn.torch.frontend.raw_ops

Combine raw ops which are not directly supported by PyTorch.

returnn.torch.frontend.raw_ops.squared_difference(a: Tensor, b: Tensor) Tensor[source]
Parameters:
  • a

  • b

Returns:

(a - b) ** 2