ext/torch/nn.cpp in torch-rb-0.9.1 vs ext/torch/nn.cpp in torch-rb-0.9.2
- old
+ new
@@ -96,11 +96,14 @@
"grad",
[](Parameter& self) {
auto grad = self.grad();
return grad.defined() ? Object(Rice::detail::To_Ruby<torch::Tensor>().convert(grad)) : Nil;
})
+ // can't use grad=
+ // assignment methods fail with Ruby 3.0
+ // TODO add checks like Tensor
.define_method(
- "grad=",
+ "_set_grad",
[](Parameter& self, torch::Tensor& grad) {
self.mutable_grad() = grad;
})
.define_singleton_function(
"_make_subclass",