returnn 1.20230418.133927__tar.gz → 1.20230418.134656__tar.gz

{returnn-1.20230418.133927/returnn.egg-info → returnn-1.20230418.134656}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: returnn
-Version: 1.20230418.133927
+Version: 1.20230418.134656
 Summary: The RWTH extensible training framework for universal recurrent neural networks
 Home-page: https://github.com/rwth-i6/returnn/
 Author: Albert Zeyer

returnn-1.20230418.134656/_setup_info_generated.py

@@ -0,0 +1,2 @@
+version = '1.20230418.134656'
+long_version = '1.20230418.134656+git.79116cf'

{returnn-1.20230418.133927 → returnn-1.20230418.134656}/returnn/torch/frontend/_backend.py

@@ -515,6 +515,8 @@ class TorchBackend(Backend[torch.Tensor]):
         :return: a `kind` b
         """
         assert a.dim() == b.dim()
+        if kind == "squared_difference":
+            return (a - b) ** 2
         kind = {
             "truediv": "true_divide",
             "floordiv": "floor_divide",

{returnn-1.20230418.133927 → returnn-1.20230418.134656/returnn.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: returnn
-Version: 1.20230418.133927
+Version: 1.20230418.134656
 Summary: The RWTH extensible training framework for universal recurrent neural networks
 Home-page: https://github.com/rwth-i6/returnn/
 Author: Albert Zeyer

{returnn-1.20230418.133927 → returnn-1.20230418.134656}/tests/test_rf_math.py

@@ -28,3 +28,25 @@ def test_neg():
         out.mark_as_default_output(shape=(batch_dim, time_dim, in_dim))
 
     run_model(extern_data, lambda *, epoch, step: _Net(), _forward_step)
+
+
+def test_squared_difference():
+    time_dim = Dim(Tensor("time", [batch_dim], dtype="int32"))
+    in_dim = Dim(7, name="in")
+    extern_data = TensorDict(
+        {
+            "a": Tensor("a", [batch_dim, time_dim, in_dim], dtype="float32"),
+            "b": Tensor("b", [batch_dim, time_dim, in_dim], dtype="float32"),
+        }
+    )
+
+    class _Net(rf.Module):
+        def __call__(self, a: Tensor, b: Tensor) -> Tensor:
+            return rf.squared_difference(a, b)
+
+    # noinspection PyShadowingNames
+    def _forward_step(*, model: _Net, extern_data: TensorDict):
+        out = model(extern_data["a"], extern_data["b"])
+        out.mark_as_default_output(shape=(batch_dim, time_dim, in_dim))
+
+    run_model(extern_data, lambda *, epoch, step: _Net(), _forward_step)

returnn-1.20230418.133927/_setup_info_generated.py

@@ -1,2 +0,0 @@
-version = '1.20230418.133927'
-long_version = '1.20230418.133927+git.434655e'