通过numpy生成数据
用户可以通过Python中numpy模块生成Tensor数据作为输入数据/标杆数据,然后调用tensor_xxx接口导入,output数据落盘文件命名为{name}_output.bin,以便区分input数据。
- 输入/输出不同地址(非原地算子)
import numpy as np import ascendebug # 生成输入/标杆数据 x = np.random.uniform(1, 100, (16384)).astype(np.float16) y = np.random.uniform(1, 100, (16384)).astype(np.float16) z = x + y # 配置输入数据 debug_op = ascendebug.create_debug_op('add_custom').tensor_input('x', x) .tensor_input('y', y).tensor_output('z', z)
- 输入/输出同地址(原地算子)
import numpy as np import ascendebug # 生成输入/标杆数据 x = np.random.uniform(1, 100, (16384)).astype(np.float16) y = np.random.uniform(1, 100, (16384)).astype(np.float16) z = x + y # 配置输入数据 debug_op = ascendebug.create_debug_op('add_custom').tensor_input('x', x) .tensor_input('y', y).tensor_output('x', z)
父主题: API方式下数据准备说明