FloorMod

产品支持情况

产品	是否支持
昇腾910_95 AI处理器	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√

功能说明

• 算子功能: 将scalar self进行broadcast成和tensor other一样shape的tensor以后，其中的每个元素都转换为除以other的对应元素以后得到的余数。该结果与除数other同符号，并且该结果的绝对值是小于other的绝对值。 实际计算remainder(self, other) 等效于以下公式：

  $$ou{t}_i=self-floor(self/othe{r}_i)\ast othe{r}_i$$

• 示例：

self = 5.0   # float
other = tensor([[-1, -2],
                [-3, -4]]).type(int32)
result = remainder(self, other)

# result的值
# tensor([[ 0., -1.],
#         [-1., -3.]])  float

# 对于元素other中的-4来说，计算结果为 5 - floor(5 / -4) * -4 = -3
# 可以看到，最终结果-3的绝对值小于原来的-4的绝对值。

• 参数说明：

  • self(aclScalar*, 计算输入)：公式中的输入self，Host侧的aclScalar。

    • 昇腾910_95 AI处理器：数据类型支持INT32、INT64、FLOAT16、FLOAT、DOUBLE、BFLOAT16。数据类型与other的数据类型需满足数据类型推导规则（参见TensorScalar互推导关系），并且推导出的数据类型必须能转换为out的数据类型。
    • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件、Atlas A3 训练系列产品/Atlas A3 推理系列产品：数据类型支持INT32、INT64、FLOAT16、FLOAT、DOUBLE、BFLOAT16。数据类型与other的数据类型需满足数据类型推导规则（参见互推导关系），并且推导出的数据类型必须能转换为out的数据类型。
    • Atlas 训练系列产品：数据类型支持INT32、INT64、FLOAT16、FLOAT、DOUBLE。数据类型与other的数据类型需满足数据类型推导规则（参见互推导关系），并且推导出的数据类型必须能转换为out的数据类型。

  • other(aclTensor*, 计算输入)：公式中的输入other, Device侧的aclTensor，支持非连续的Tensor，数据格式支持ND，数据维度不支持8维以上。

    • 昇腾910_95 AI处理器：数据类型支持INT32、INT64、FLOAT16、FLOAT、DOUBLE、BFLOAT16。数据类型与self的数据类型需满足数据类型推导规则（参见TensorScalar互推导关系），并且推导出的数据类型必须能转换为out的数据类型。
    • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件、Atlas A3 训练系列产品/Atlas A3 推理系列产品：数据类型支持INT32、INT64、FLOAT16、FLOAT、DOUBLE、BFLOAT16。数据类型与self的数据类型需满足数据类型推导规则（参见互推导关系），并且推导出的数据类型必须能转换为out的数据类型。
    • Atlas 训练系列产品：数据类型支持INT32、INT64、FLOAT16、FLOAT、DOUBLE。数据类型与self的数据类型需满足数据类型推导规则（参见互推导关系），并且推导出的数据类型必须能转换为out的数据类型。

  • out(aclTensor *, 计算输出)：公式中的输出out，Device侧的aclTensor。shape需要与other一致。支持非连续的Tensor，数据格式支持ND。

    • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件、Atlas A3 训练系列产品/Atlas A3 推理系列产品、昇腾910_95 AI处理器：数据类型支持INT32、INT64、FLOAT16、FLOAT、DOUBLE、BFLOAT16。
    • Atlas 训练系列产品：数据类型支持INT32、INT64、FLOAT16、FLOAT、DOUBLE。

  • workspaceSize(uint64_t *，出参)：返回需要在Device侧申请的workspace大小。

  • executor(aclOpExecutor **，出参)：返回op执行器，包含了算子计算流程。

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_remainder_scalar_tensor	通过aclnnRemainderScalarTensor接口方式调用FloorDiv算子