flatten¶

dragon.flatten(
  inputs,
  axis=0,
  end_axis=- 1,
  copy=True,
  **kwargs
)[source]¶

Flatten the input along the given axes.

Set keep_axes to flatten if shape is dynamic.

Examples:

x = dragon.ones((1, 2, 3, 4))
print(dragon.flatten(x, axis=1).shape)  # (1, 24)
print(dragon.flatten(x, axis=1, end_axis=2).shape)  # (1, 6, 4)

Parameters:

• inputs (dragon.Tensor) – The input tensor.
• axis (int, optional, default=0) – The first axis to flatten.
• end_axis (int, optional, default=-1) – The last axis to flatten.
• copy (bool, optional, default=True) – Return a new tensor or call in-place.

Returns:

dragon.Tensor – The output tensor.