shape_to_list
函数功能
将tvm格式的shape转换为列表格式的shape。
函数原型
def shape_to_list(shape)
参数说明
参数 |
说明 |
---|---|
shape |
数据shape |
返回值说明
返回列表。
约束说明
无
调用示例
from tbe.common.utils import shape_util shape_util.shape_to_list((32,64,64,3))
把shape(32,64,64,3)转为列表[32,64,64,3]。
父主题: shape相关工具