lib/tensorflow/math.rb in tensorflow-0.1.2 vs lib/tensorflow/math.rb in tensorflow-0.2.0

@@ -491,10 +491,10 @@
 
       private
 
       def reduction_dims(input_tensor)
         rank = RawOps.rank(input: input_tensor).value
-        (0...rank).to_a
+        TensorFlow.constant((0...rank).to_a, dtype: :int32)
       end
     end
   end
 end