下载
中文
注册

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"。

  • 若options=["optional"],表示本输入是可选的。
  • 若options=["ignore"],表示本输入是可忽略的。

返回值

返回DebugOp类型的自身实例对象。

使用说明

import ascendebug
ascendebug.create_debug_op('OpCustom', 'VectorCore', '${chip_version}').custom_output('z', 'float16', [24, 144, 1280],'/xxxx/z.bin')