lib/torch/nn/batch_norm.rb in torch-rb-0.2.3 vs lib/torch/nn/batch_norm.rb in torch-rb-0.2.4

@@ -68,8 +68,13 @@
           weight: @weight, bias: @bias,
           training: @training || !@track_running_stats,
           momentum: exponential_average_factor, eps: @eps
         )
       end
+
+      def extra_inspect
+        s = "%{num_features}, eps: %{eps}, momentum: %{momentum}, affine: %{affine}, track_running_stats: %{track_running_stats}"
+        format(s, **dict)
+      end
     end
   end
 end