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

Tensor构造函数

函数功能

用于构造指定的Tensor对象。Tensor类用来描述一个tensor对象的信息以及行为,包含:shape信息、format信息、datatype信息以及tensor数据内容tensordata。

函数原型

Tensor (const StorageShape &storage_shape, const StorageFormat &storage_format, const TensorPlacement placement, const ge::DataType data_type, TensorAddress addr)

Tensor(const StorageShape &storage_shape, const StorageFormat &storage_format, ge::DataType data_type)

参数说明

参数

输入/输出

说明

storage_shape

输入

指定tensor的shape信息

storage_format

输入

指定tensor的format信息

placement

输入

指定tensor的实际数据所存储的device位置

data_type

输入

指定tensor的datatype信息

addr

输入

指定tensor的实际数据所存储的内存地址

返回值说明

返回一个初始化的Tensor对象。

约束说明

Tensor中的成员(format、shape等)的值均需要显示设置,否则均为未定义值。

调用示例

Tensor tensor{{{8, 3, 224, 224}, {16, 3, 224, 224}},       // shape                
              {ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}},  // format                              
              kFollowing,                                  // placement                
              ge::DT_FLOAT16,                              //dt                
              nullptr};
搜索结果
找到“0”个结果

当前产品无相关内容

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