Use none as a default for optional tensors rather than an empty tensor.

LaurentMazare · Sep 11, 2024 · 51b966b · 51b966b
1 parent 8647136
commit 51b966b
Show file tree

Hide file tree

Showing 2 changed files with 197 additions and 197 deletions.
diff --git a/gen/gen.ml b/gen/gen.ml
@@ -229,7 +229,7 @@ module Func = struct
            c10::optional<at::Layout>(static_cast<at::Layout>(%s)))"
           arg_name
           arg_name
-      | TensorOption -> Printf.sprintf "(%s ? *%s : torch::Tensor())" arg_name arg_name
+      | TensorOption -> Printf.sprintf "(%s ? ::std::optional<at::Tensor>(*%s) : ::std::nullopt)" arg_name arg_name
       | Bool -> "(bool)" ^ arg_name
       | IntList -> Printf.sprintf "torch::IntArrayRef(%s_data, %s_len)" arg_name arg_name
       | IntListOption ->