昇腾社区首页
中文
注册

SetBatchNum

功能说明

在不改变tiling的情况下,重新设置多Batch计算的Batch数。

函数原型

__aicore__ inline void SetBatchNum(int32_t batchA, int32_t batchB)

参数说明

表1 参数说明

参数名

输入/输出

描述

batchA

输入

设置的一次计算的A矩阵batch数。

batchB

输入

设置的一次计算的B矩阵batch数。

返回值

注意事项