FloorMod

接口功能

计算两个输入张量的逐元素模运算

函数原型

1
2
3
4
5
int FloorMod(T *input, T *mod, T *output, size_t length);            //input,mod均为指针变量

int FloorMod(T input, T *mod, T *output, size_t length);             //input为指针变量

int FloorMod(T *input, T mod, T *output, size_t length);             //mod为指针变量
  • 该接口只接受一维张量作为输入。
  • 模板类型T:支持float,double。
  • 指针有效长度由用户决定。
  • 保证输入指针的有效空间相互不重叠。

参数说明

表1 参数说明

参数

输入/输出

说明

约束

input

输入

被取模运算的张量。

  • 指针不为空,且只接受一维张量。或为标量,不可与mod同时为标量。
  • 矩阵中不含有nan或者inf,需用户保障,防止计算结果未定义

mod

输入

用于取模运算的除数。

  • 指针不为空,且只接受一维张量。或为标量,不可与input同时为标量。
  • 元素中不含有0,需用户保障,防止计算结果未定义。输入为矩阵时,元素不含有nan或者inf,需用户保障,防止计算结果未定义

output

输入

用于存储计算结果,构成的张量作为输出结果。

指针不为空,且只接受一维张量。

length

输入

用于计算的张量长度。

取值范围[0,2^64-1],length等于output的有效空间,input有效空间,mod有效空间的最小值,需用户保障,防止缓冲区溢出等安全问题。