documentation

Author: xadupre

CHANGELOGS.rst

@@ -4,6 +4,8 @@ Change Logs
 0.7.17
 ++++++
 
+* :pr:`275`: fixes function ``patched_vmap``
+
 0.7.16
 ++++++
 

onnx_diagnostic/export/shape_helper.py

@@ -212,6 +212,23 @@ def make_fake_with_dynamic_dimensions(
     constraints as the following dynamic shapes.
     This uses function :func:`onnx_diagnostic.helpers.fake_tensor_helper.make_fake`.
 
+    A simple tensor:
+
+    .. runpython::
+        :showcode:
+
+        import torch
+        from onnx_diagnostic.helpers.cache_helper import make_dynamic_cache
+        from onnx_diagnostic.export.shape_helper import make_fake_with_dynamic_dimensions
+
+        inputs, _ = make_fake_with_dynamic_dimensions(
+            torch.rand((2, 3, 4, 5), dtype=dtype=torch.float32),
+            {0: "batch", 2: "cache_length"},
+        )
+        print(inputs)
+
+    With a cache:
+
     .. runpython::
         :showcode: