custom_output
函数功能
向DebugOp对象中增加bin文件存储的输出描述。
函数原型
def custom_output(self, name: str, dtype: str, shape: List, data_file: str, options=None)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
name |
输入 |
字符串,输出描述的name。 |
dtype |
输入 |
数据类型。 |
shape |
输入 |
输入数据的shape,例如[1,1024]。 |
data_file |
输入 |
输入数据的文件名称。 |
options |
输入 |
其他可选参数配置,list格式,可选填"optional"、"ignore"。
|
返回值
返回DebugOp类型的自身实例对象。
使用说明
import ascendebug ascendebug.create_debug_op('OpCustom', 'VectorCore', '${chip_version}').custom_output('z', 'float16', [24, 144, 1280],'/xxxx/z.bin')
父主题: DebugOp类