下载
中文
注册

list_tensor_input

函数功能

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

函数原型

def list_tensor_input(self, input_desc_list: List)

参数说明

参数名

输入/输出

说明

input_desc_list

输入

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

返回值

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

使用说明

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