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

# Slice

从输入张量中提取一个子集（切片）。根据给定的起始索引（begin）和切片大小（size），在每个维度上截取数据。

\[output[i_0, i_1, \dots, i_{n-1}] = input[begin_0 + i_0, begin_1 + i_1, \dots, begin_{n-1} + i_{n-1}]\]

其中 $n$ 为维度数（ndim），且满足 $0 le i_k < size_k$。

输入：

  - input - 输入张量数据地址。

  - input_shape - 输入张量的形状数组地址。

  - ndim - 输入张量的维度数。

  - begin - 切片起始索引数组地址（长度为 ndim）。

  - size - 切片大小数组地址（长度为 ndim）。

  - core_mask(int, 可选) - 核掩码（仅适用于共享存储版本）。

输出：

  - output - 计算结果存储地址（输出张量形状即为 size）。

支持平台：

  `FT78NE` `MT7004`

备注

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

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

- 切片操作不改变数据数值，仅改变数据的空间排布，常用于特征提取或张量分解。

共享存储版本:

void i8_slice_s(int8_t *input, int8_t *output, int *input_shape, int ndim, int *begin, int *size, int core_mask)

void i16_slice_s(int16_t *input, int16_t *output, int *input_shape, int ndim, int *begin, int *size, int core_mask)

void i32_slice_s(int32_t *input, int32_t *output, int *input_shape, int ndim, int *begin, int *size, int core_mask)

void hp_slice_s(half *input, half *output, int *input_shape, int ndim, int *begin, int *size, int core_mask)

void fp_slice_s(float *input, float *output, int *input_shape, int ndim, int *begin, int *size, int core_mask)

void dp_slice_s(double *input, double *output, int *input_shape, int ndim, int *begin, int *size, int core_mask)

void c64_slice_s(float *input, float *output, int *input_shape, int ndim, int *begin, int *size, int core_mask)

void c128_slice_s(double *input, double *output, int *input_shape, int ndim, int *begin, int *size, int core_mask)

  C调用示例：

       1// FT78NE 示例（多核并行切片）
       2#include <stdio.h>
       3#include "78NE/utils.h"
       4
       5int main() {
       6    float *input = (float *)0xA0000000;
       7    float *output = (float *)0xB0000000;
       8    int input_shape[] = {16, 32, 64, 128};
       9    int begin[] = {2, 4, 8, 16};
      10    int size[] = {8, 12, 20, 32};
      11    int ndim = 4;
      12    int core_mask = 0xFF; // 使用8核并行
      13
      14    fp_slice_s(input, output, input_shape, ndim, begin, size, core_mask);
      15    return 0;
      16}

私有存储版本:

void i8_slice_p(int8_t *input, int8_t *output, int *input_shape, int ndim, int *begin, int *size)

void i16_slice_p(int16_t *input, int16_t *output, int *input_shape, int ndim, int *begin, int *size)

void i32_slice_p(int32_t *input, int32_t *output, int *input_shape, int ndim, int *begin, int *size)

void hp_slice_p(half *input, half *output, int *input_shape, int ndim, int *begin, int *size)

void fp_slice_p(float *input, float *output, int *input_shape, int ndim, int *begin, int *size)

void dp_slice_p(double *input, double *output, int *input_shape, int ndim, int *begin, int *size)

void c64_slice_p(float *input, float *output, int *input_shape, int ndim, int *begin, int *size)

void c128_slice_p(double *input, double *output, int *input_shape, int ndim, int *begin, int *size)

  C调用示例：

       1// MT7004 示例（单核私有存储切片）
       2#include <stdio.h>
       3
       4int main() {
       5    float *input = (float *)0x10000000;
       6    float *output = (float *)0x10010000;
       7    int input_shape[] = {4, 8, 16, 32};
       8    int begin[] = {1, 2, 3, 4};
       9    int size[] = {2, 3, 6, 8};
      10    int ndim = 4;
      11
      12    fp_slice_p(input, output, input_shape, ndim, begin, size);
      13    return 0;
      14}