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"