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