AllocTensorMsg

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	x
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 200I/500 A2 推理产品	x
Atlas 推理系列产品	x
Atlas 训练系列产品	x

根据shape和data type申请Tensor类型的msg。该函数供Proc调用。

std::shared_ptr<FlowMsg> AllocTensorMsg(const std::vector<int64_t> &shape, TensorDataType dataType)

参数名	输入/输出	描述
Shape	输入	Tensor的Shape。
dataType	输入	Tensor的dataType。

申请的Tensor指针。

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

无。

父主题： MetaRunContext类