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",