Simplified code

ankane · Nov 11, 2024 · e3b38de · e3b38de
1 parent a1b7b77
commit e3b38de
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/ext/torch/torch.cpp b/ext/torch/torch.cpp
@@ -16,13 +16,7 @@ torch::Tensor make_tensor(Rice::Array a, const std::vector<int64_t> &size, const
     vec.push_back(Rice::detail::From_Ruby<T>().convert(a[i].value()));
   }
 
-  // hack for requires_grad error
-  auto requires_grad = options.requires_grad();
-  torch::Tensor t = torch::tensor(vec, options.requires_grad(c10::nullopt));
-  if (requires_grad) {
-    t.set_requires_grad(true);
-  }
-
+  torch::Tensor t = torch::tensor(vec, options);
   return t.reshape(size);
 }