Tensor构造函数

函数功能

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

函数原型

Tensor() //默认构造函数

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)

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

参数说明

1
2
3
4
5
6
7

using TensorAddrManager = ge::graphStatus (*)(TensorAddress addr, TensorOperateType operate_type, void **out);
enum TensorOperateType {
  kGetTensorAddress,  ///< 获取Tensor的地址  
  kFreeTensor,        ///< 释放Tensor  
  kPlusShareCount,    ///< 共享Tensor  
  kTensorOperateType
};

返回值说明

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

约束说明

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

调用示例

1
2
3
4
5

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};