shape_to_list
Description
Converts shape from the format of tvm to list.
Prototype
def shape_to_list(shape)
Parameters
Parameter |
Description |
|---|---|
shape |
Shape. |
Returns
A list
Restrictions
None
Example
from tbe.common.utils import shape_util shape_util.shape_to_list((32,64,64,3))
Converts shape (32,64,64,3) to a list [32,64,64,3].
Parent topic: Shape-related Tools