多维张量的规约计算方法、装置、设备、存储介质和产品

本发明实施方式公开了多维张量的规约计算方法、装置、设备、存储介质和产品。方法包括：确定n维张量的m个规约维度和n‑m个非规约维度；生成规约计算指令，规约计算指令被执行后在人工智能芯片上启动K个线程块，K个线程块在m个规约维度上以并行方式进行规约计算；每个线程块包含的线程数与m个规约维度所包含的元素数目相关联，K与n‑m个非规约维度所包含的元素数目相关联，每个线程块中的各个线程以并行方式读取各自的规约数据；向人工智能芯片发送规约计算指令，人工智能芯片执行规约计算指令。可以针对多维张量的任意维度进行规约计算，提高通用性，减少计算时间，还提高计算效率。