# Sqrtgrad — MindSpore Signal+ 使用手册 alpha 文档

# Sqrtgrad

计算 Sqrt（平方根）操作的梯度。该算子是 Sqrt 算子的反向传播（backward pass）部分。

\[\text{output}_i = \frac{1}{2} \times \frac{\text{input2}_i}{\text{input1}_i}\]

其中 input1 是前向传播时 Sqrt 的输出（即 \(y = \sqrt{x}\)），input2 是来自后一层的上游梯度 \(dy\)，output 是对原始输入 \(x\) 的梯度 \(dx\)。

输入：

  - input1 - 前向传播时 Sqrt 的输出数据地址（即 \(y = \sqrt{x}\)）。

  - input2 - 来自后一层的上游梯度数据地址（即 \(dy\)）。

  - size - 计算长度。

  - core_mask - 核掩码（仅共享存储版本需要）。

输出：

  - output - 计算出的对原始输入的梯度数据地址（即 \(dx\)）。

支持平台：

  `FT78NE` `MT7004`

备注

- FT78NE 支持fp32, int8, int16, int32, fp64, cplx64, cplx128

- MT7004 支持fp16, fp32, int16, int32, cplx64

共享存储版本:

void i8_sqrtgrad_s(int8_t *input1, int8_t *input2, float *output, int size, int core_mask)

void i16_sqrtgrad_s(int16_t *input1, int16_t *input2, float *output, int size, int core_mask)

void i32_sqrtgrad_s(int32_t *input1, int32_t *input2, float *output, int size, int core_mask)

void hp_sqrtgrad_s(half *input1, half *input2, half *output, int size, int core_mask)

void fp_sqrtgrad_s(float *input1, float *input2, float *output, int size, int core_mask)

void dp_sqrtgrad_s(double *input1, double *input2, double *output, int size, int core_mask)

void c64_sqrtgrad_s(float *input1, float *input2, float *output, int size, int core_mask)

void c128_sqrtgrad_s(double *input1, double *input2, double *output, int size, int core_mask)

C调用示例：

     1//FT78NE示例
     2#include <stdio.h>
     3#include <sqrtgrad.h>
     4
     5int main(int argc, char* argv[]) {
     6    // 假设在DDR空间
     7    float *input1 = (float *)0xA0000000;   // Sqrt的输出 y = sqrt(x)
     8    float *input2 = (float *)0xA1000000;   // 上游梯度 dy
     9    float *output = (float *)0xB0000000;   // 输出梯度 dx
    10
    11    int size = 1000;
    12    int core_mask = 0xff;
    13
    14    fp_sqrtgrad_s(input1, input2, output, size, core_mask);
    15    return 0;
    16}

私有存储版本:

void i8_sqrtgrad_p(int8_t *input1, int8_t *input2, float *output, int size)

void i16_sqrtgrad_p(int16_t *input1, int16_t *input2, float *output, int size)

void i32_sqrtgrad_p(int32_t *input1, int32_t *input2, float *output, int size)

void hp_sqrtgrad_p(half *input1, half *input2, half *output, int size)

void fp_sqrtgrad_p(float *input1, float *input2, float *output, int size)

void dp_sqrtgrad_p(double *input1, double *input2, double *output, int size)

void c64_sqrtgrad_p(float *input1, float *input2, float *output, int size)

void c128_sqrtgrad_p(double *input1, double *input2, double *output, int size)

C调用示例：

     1//FT78NE示例
     2#include <stdio.h>
     3#include <sqrtgrad.h>
     4
     5int main(int argc, char* argv[]) {
     6    // 假设在L2空间
     7    float *input1 = (float *)0x10000000;   // Sqrt的输出 y = sqrt(x)
     8    float *input2 = (float *)0x10001000;   // 上游梯度 dy
     9    float *output = (float *)0x10002000;   // 输出梯度 dx
    10
    11    int size = 1000;
    12
    13    fp_sqrtgrad_p(input1, input2, output, size);
    14    return 0;
    15}