shape¶

dragon.shape(
  inputs,
  **kwargs
)[source]¶

Return the shape of input.

Examples:

x = dragon.ones((2, 3))
print(x.shape)  # Return a sequence
print(dragon.shape(x))  # Return a tensor

Parameters:

• inputs (dragon.Tensor) – The input tensor.

Returns:

dragon.Tensor – The tensor shape.