reshape¶

dragon.vm.tensorflow.reshape(
  tensor,
  shape,
  copy=True,
  name=None
)[source]¶

Change the dimensions of input.

Examples:

# Provide a determined value for each dimension if possible
x = tf.ones(shape=(1, 2, 3, 4))
print(tf.reshape(x, shape=[6, 4]).shape)  # [6, 4]

# Set the existing dimensions to ``0`` if it unchanged
print(tf.reshape(x, shape=[0, 0, 12]).shape)  # [1, 2, 12]
print(tf.reshape(x, shape=[0, 0, 0, 0]).shape)  # [1, 2, 3, 4]
print(tf.reshape(x, shape=[0, 0, 0, 0, 0]).shape)  # Wrong

# You can also set ``-1`` once to infer the value
print(tf.reshape(x, shape=[-1, 4]).shape)  # [6, 4]
print(tf.reshape(x, shape=[-1, -1]).shape)  # Wrong

Parameters:

• tensor (dragon.Tensor) – The input tensor.
• shape (Union[Sequence[int], dragon.Tensor]) – The output shape.
• copy (bool, optional, default=True) – Return a new tensor or call in-place.
• name (str, optional) – The operation name.