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();