ext/torch/ext.cpp in torch-rb-0.3.5 vs ext/torch/ext.cpp in torch-rb-0.3.6

- old
+ new

@@ -300,15 +300,10 @@ std::copy(s.begin(), s.end(), std::back_inserter(v)); // https://github.com/pytorch/pytorch/issues/20356#issuecomment-567663701 return torch::pickle_load(v); }) .define_singleton_method( - "_binary_cross_entropy_with_logits", - *[](const Tensor &input, const Tensor &target, OptionalTensor weight, OptionalTensor pos_weight, MyReduction reduction) { - return torch::binary_cross_entropy_with_logits(input, target, weight, pos_weight, reduction); - }) - .define_singleton_method( "_from_blob", *[](String s, IntArrayRef size, const torch::TensorOptions &options) { void *data = const_cast<char *>(s.c_str()); return torch::from_blob(data, size, options); }) @@ -376,14 +371,9 @@ }) .define_method( "_requires_grad!", *[](Tensor& self, bool requires_grad) { return self.set_requires_grad(requires_grad); - }) - .define_method( - "_backward", - *[](Tensor& self, OptionalTensor gradient, bool create_graph, bool retain_graph) { - return self.backward(gradient, create_graph, retain_graph); }) .define_method( "grad", *[](Tensor& self) { auto grad = self.grad();