reshape

Graph.reshape(tensor, shape, *, name=None)

Reshape a tensor into a new shape, keeping the order of its elements.

Parameters

  • tensor (np.ndarray or Tensor) – The tensor you want to reshape.
  • shape (tuple [ int , ... ]) – The new shape of the tensor.
  • name (str or None , optional) – The name of the node.

Returns

The reshaped tensor.

Return type

Tensor

Was this useful?