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