list_tensor_output

函数功能

向DebugOp对象中增加list tensor类型的输出描述,每个list tensor中均是tensor类型的元素。

函数原型

def list_tensor_output(self, output_desc_list: List)

参数说明

参数名

输入/输出

说明

output_desc_list

输入

list,其中每个元素类型为tensor_output中的入参组成的tuple。

返回值

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

使用说明

import numpy as np
import ascendebug
z_0 = np.random.uniform(1, 100, (16384)).astype(np.float16)
z_1 = np.random.uniform(1, 100, (16384)).astype(np.float16)
ascendebug.create_debug_op('OpCustom', 'VectorCore', '${chip_version}').list_tensor_output([('z_0', z_0, []), ('z_1', z_1, [])])