lib/torch/nn/parameter.rb in torch-rb-0.5.3 vs lib/torch/nn/parameter.rb in torch-rb-0.6.0
- old
+ new
@@ -1,10 +1,10 @@
module Torch
module NN
class Parameter < Tensor
def self.new(data = nil, requires_grad: true)
data = Tensor.new unless data
- Tensor._make_subclass(data, requires_grad)
+ _make_subclass(data, requires_grad)
end
def inspect
"Parameter containing:\n#{super}"
end