Merge pull request #891 from LaurentMazare/opt-tensor-fix

Use none as a default for optional tensors rather than an empty tensor.
LaurentMazare · Sep 12, 2024 · 2687871 · 2687871
2 parents 10ef423 + 51b966b
commit 2687871
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 ->