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类