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, [])])
父主题: DebugOp类