lib/torch/nn/parameter.rb in torch-rb-0.1.4 vs lib/torch/nn/parameter.rb in torch-rb-0.1.5

@@ -3,8 +3,12 @@
     class Parameter < Tensor
       def self.new(data = nil, requires_grad: true)
         data = Tensor.new unless data
         Tensor._make_subclass(data, requires_grad)
       end
+
+      def inspect
+        "Parameter containing:\n#{super}"
+      end
     end
   end
 end