lib/torch/nn/linear.rb in torch-rb-0.5.3 vs lib/torch/nn/linear.rb in torch-rb-0.6.0

@@ -1,7 +1,9 @@
 module Torch
   module NN
     class Linear < Module
+      attr_reader :in_features, :out_features
+
       def initialize(in_features, out_features, bias: true)
         super()
         @in_features = in_features
         @out_features = out_features