lib/torch/native/parser.rb in torch-rb-0.2.5 vs lib/torch/native/parser.rb in torch-rb-0.2.6
- old
+ new
@@ -73,10 +73,10 @@
case t
when "Tensor"
v.is_a?(Tensor)
when "Tensor?"
v.nil? || v.is_a?(Tensor)
- when "Tensor[]"
+ when "Tensor[]", "Tensor?[]"
v.is_a?(Array) && v.all? { |v2| v2.is_a?(Tensor) }
when "int"
if k == "reduction"
v.is_a?(String)
else