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.