codegen/generate_functions.rb in torch-rb-0.7.0 vs codegen/generate_functions.rb in torch-rb-0.8.0

- old
+ new

@@ -323,10 +323,12 @@ case param[:type] when "Tensor" "tensor" when "Tensor[]" "tensorlist" + when "Scalar[]" + "scalarlist" when /\Aint\[/ "intlist" when "float[]" "doublelist" when "Scalar" @@ -412,10 +414,12 @@ else "const Tensor &" end when "Tensor[]" "TensorList" + when "Scalar[]" + "ScalarList" when "int" "int64_t" when "float" "double" when /\Aint\[/ @@ -530,9 +534,11 @@ case param[:type] when "Tensor", /\ATensor\([a-z]!?\)\z/ "Tensor" when /\Tensor\[\d*\]\z/ "TensorList" + when "Scalar[]" + "ScalarList" when /\ADimname\[\d*\]\z/ "DirnameList" when /\Aint\[\d*\]\z/ "IntArrayRef" when "int"