lib/torch/nn/parameter.rb in torch-rb-0.8.2 vs lib/torch/nn/parameter.rb in torch-rb-0.8.3
- old
+ new
@@ -7,8 +7,14 @@
end
def inspect
"Parameter containing:\n#{super}"
end
+
+ def dup
+ Torch.no_grad do
+ Parameter.new(clone, requires_grad: requires_grad)
+ end
+ end
end
end
end