aclnnNLLLoss2dBackward
支持的产品型号
- Atlas 训练系列产品。
- Atlas A2训练系列产品。
接口原型
每个算子分为，必须先调用“aclnnNLLLoss2dBackwardGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnNLLLoss2dBackward”接口执行计算。
aclnnStatus aclnnNLLLoss2dBackwardGetWorkspaceSize(const aclTensor *gradOutput, const aclTensor *self, const aclTensor *target, const aclTensor *weight, int64_t reduction, int64_t ignoreIndex, const aclTensor *totalWeight, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)aclnnStatus aclnnNLLLoss2dBackward(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
功能描述
算子功能：负对数似然损失反向。
aclnnNLLLoss2dBackwardGetWorkspaceSize
参数说明：
gradOutput(aclTensor*, 计算输入)：Device侧的aclTensor，shape为三维（第一维是N）或者一维（且元素个数为1），数据类型支持FLOAT、FLOAT16、BFLOAT16（仅Atlas A2训练系列产品支持）。支持，支持ND。
self(aclTensor*, 计算输入)：Device侧的aclTensor，shape为四维，第一维是N表示batch size，第二维是C表示类别，数据类型支持FLOAT、FLOAT16、BFLOAT16（仅Atlas A2训练系列产品支持）。支持，支持ND。
target(aclTensor*, 计算输入)：Device侧的aclTensor，表示真实标签，shape为三维，第一维是N，其中每个元素的取值范围是[0, C - 1]，数据类型支持INT64、UINT8。支持，支持ND。
weight(aclTensor*, 计算输入)：Device侧的aclTensor，表示各个类别的权重，shape为(C, )，数据类型支持FLOAT、FLOAT16。支持，支持ND。
reduction(int64_t, 计算输入)：Host侧的int64_t，指定要应用到输出的缩减。支持 0('none') | 1('mean') | 2('sum')。'none' 表示不应用减少，'mean' 表示输出的总和将除以输出中的元素数，'sum' 表示输出将被求和。
ignoreIndex(int64_t, 计算输入)：Host侧的int64_t，指定一个被忽略且不影响输入梯度的目标值。
totalWeight(aclTensor*, 计算输入)：Device侧的aclTensor，数据类型支持FLOAT、FLOAT16，且数据类型为与weight相同，shape为(1，)，支持ND。
out(aclTensor*, 计算输出)：Device侧的aclTensor，shape与self相同。支持，支持ND。
workspaceSize(uint64_t*, 出参)：返回用户需要在Device侧申请的workspace大小。
executor(aclOpExecutor**, 出参)：返回op执行器，包含了算子计算流程。
返回值：
aclnnNLLLoss2dBackward
参数说明：
workspace(void*, 入参)：在Device侧申请的workspace内存地址。
workspaceSize(uint64_t, 入参)：在Device侧申请的workspace大小，由第一段接口aclnnNLLLoss2dBackwardGetWorkspaceSize获取。
executor(aclOpExecutor*, 入参)：op执行器，包含了算子计算流程。
stream(aclrtStream, 入参)：指定执行任务的 AscendCL Stream流。
返回值：
约束与限制
无。