lib/tensorflow/ops.rb in tensorflow-0.1.2 vs lib/tensorflow/ops.rb in tensorflow-0.2.0
- old
+ new
@@ -27,9 +27,13 @@
start = 0
end
RawOps.range(start: start, limit: limit, delta: delta)
end
+ def reshape(tensor, shape)
+ RawOps.reshape(tensor: tensor, shape: shape)
+ end
+
def squeeze(input, axis: nil)
RawOps.squeeze(input: input, squeeze_dims: axis)
end
def timestamp