下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

AllocTensorListMsg

函数功能

根据输入的dtype shapes数组输入一块连续内存,用于承载tensor数组。

函数原型

virtual std::shared_ptr<FlowMsg> AllocTensorListMsg(const std::vector<std::vector<int64_t>> &shapes, const std::vector<TensorDataType> &dataTypes, uint32_t align = 512U)

参数说明

参数名

输入/输出

描述

shapes

输入

Tensor List对应的shape列表。

dataTypes

输入

Tensor List的dataType列表。

align

输入

申请内存地址对齐大小,取值范围 【32、64、128、256、512、1024】。

返回值

申请的FlowMsg指针。

异常处理

申请不到tensor指针则返回NULL。

约束说明

无。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词