42 lines
1.4 KiB
ReStructuredText
42 lines
1.4 KiB
ReStructuredText
Squeeze
|
||
=================
|
||
|
||
返回删除指定axis中大小为1的维度后的Tensor。由于该算子仅改变张量形状，因此其DSP算子的作用是将数据从输入张量完整拷贝到输出张量。
|
||
|
||
输入：
|
||
- **src** - 输入地址
|
||
- **total_copy_size** - 计算得到的总共需拷贝的数据量，单位为字节。
|
||
- **core_mask** - 核掩码。
|
||
|
||
输出：
|
||
- **dst** - 输出地址。
|
||
|
||
支持平台：
|
||
``FT78NE``
|
||
``MT7004``
|
||
|
||
.. note::
|
||
- FT78NE 支持int8, int16, int32, fp32, fp64, cplx64, cplx128
|
||
- MT7004 支持fp16, fp32, int16, int32, cplx64
|
||
|
||
**共享/私有存储版本:**
|
||
|
||
.. c:function:: void anytype_squeeze_anycore(void* src, void* dst, int total_copy_size, int core_mask)
|
||
|
||
各种数据类型、私有及共享空间版本均使用该函数。
|
||
|
||
**C调用示例：**
|
||
|
||
.. code-block:: c
|
||
:linenos:
|
||
:emphasize-lines: 8
|
||
|
||
void main(){
|
||
int core_mask = 0b1111; // 测试单核时核掩码设置为0b0001即可
|
||
int core_num = GetCoreNum(core_mask);
|
||
float* src = (float*)0x88000000; // 测试私有空间时地址设置在私有空间内即可
|
||
float* dst = (float*)0x98000000;
|
||
int shape[3] = {1, 10, 10};
|
||
int total_copy_size = shape[0] * shape[1] * shape[2] * sizeof(float);
|
||
anytype_squeeze_anycore(src, dst, total_copy_size, core_mask);
|
||
} |