aclnnGridSampler3D
支持的产品型号
- Atlas 训练系列产品。
- Atlas A2训练系列产品/Atlas 800I A2推理产品。
接口原型
- 每个算子分为，必须先调用“aclnnGridSampler3DGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnGridSampler3D”接口执行计算。
aclnnStatus aclnnGridSampler3DGetWorkspaceSize(const aclTensor *input, const aclTensor *grid, int64_t interpolationMode, int64_t paddingMode, bool alignCorners, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)aclnnStatus aclnnGridSampler3D(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
功能描述
- 提供一个输入张量input，以及一个对应的grid网格，根据grid中每个位置提供的坐标信息，将input中对应位置的像素值填充到grid指定的位置，得到最终的输出。输入input、grid网格、输出output的尺寸如下：
aclnnGridSampler3DGetWorkspaceSize
参数说明：
- input(const aclTensor*, 计算输入)：Device侧的aclTensor，数据类型支持FLOAT16、FLOAT、DOUBLE。支持，支持NCDHW、NDHWC、ND，如果数据格式为ND则会按照NCDHW格式进行处理。
- grid(const aclTensor*, 计算输入)：Device侧的aclTensor，指定输入归一化的采样像素位置，支持shape为
(N,D_{out},H_{out},W_{out},3)，数据类型支持FLOAT16、FLOAT、DOUBLE，且数据类型与input数据类型一致。支持。 - interpolationMode(int64_t, 计算输入)：int64_t类型，表示计算输出的插值模式，支持0('bilinear') | 1('nearest')。
- paddingMode(int64_t, 计算输入)：int64_t类型，表示grid范围外填充模式。即当grid有超过[-1, 1]范围的值，则按照paddingMode定义的方式处理相应的输出，支持0('zeros') | 1('border') | 2('reflection')。
- alignCorners(bool, 计算输入)：bool类型，表示对齐角像素点的方式。如果为True，则将极值-1和1视为参考输入的角像素点的中心点。如果为False，则视为参考输入的角像素点的角点。
- out(aclTensor*, 计算输出)：Device侧的aclTensor，数据类型支持FLOAT16、FLOAT、DOUBLE，且数据类型与input的数据类型一致。支持，支持NCDHW、NDHWC、ND。
- workspaceSize(uint64_t*, 出参)：返回用户需要在Device侧申请的workspace大小。
- executor(aclOpExecutor**, 出参)：返回op执行器，包含了算子计算流程。
返回值：
[object Object]
aclnnGridSampler3D
参数说明：
- workspace(void*, 入参)：在Device侧申请的workspace内存地址。
- workspaceSize(uint64_t, 入参)：在Device侧申请的workspace大小，由第一段接口aclnnGridSampler3DGetWorkspaceSize获取。
- executor(aclOpExecutor*, 入参)：op执行器，包含了算子计算流程。
- stream(aclrtStream, 入参)：指定执行任务的 AscendCL Stream流。
返回值：
约束与限制
无
调用示例
[object Object]