lib/torch/native/function.rb in torch-rb-0.3.5 vs lib/torch/native/function.rb in torch-rb-0.3.6

@@ -84,9 +84,13 @@
 
       def ret_size
         @ret_size ||= func.split("->").last.split(", ").size
       end
 
+      def ret_array?
+        @ret_array ||= func.split("->").last.include?('[]')
+      end
+
       def out?
         out_size > 0 && base_name[-1] != "_"
       end
 
       def ruby_name