传入元素个数和数据类型,创建Tensor
static std::unique_ptr<uint8_t[]> CreateFollowing(const int64_t shape_size, const ge::DataType dt, size_t &total_size)
传入数据类型和Tensor长度,创建Tensor
static std::unique_ptr<uint8_t[]> CreateFollowing(const ge::DataType dt, const size_t tensor_size, size_t &total_size)
表1 参数说明(传入元素个数和数据类型,创建Tensor)
| 参数 | 输入/输出 | 说明 |
|---|---|---|
| shape_size | 输入 | 元素个数。 |
| dt | 输入 | 数据类型,DataType类型。 |
| total_size | 输出 | 创建出的Tensor在内存中的长度。包含Tensor对象的长度和Tensor数据的长度。 |
表2 参数说明(传入数据类型和Tensor长度,创建Tensor)
| 参数 | 输入/输出 | 说明 |
|---|---|---|
| dt | 输入 | 数据类型,DataType类型。 |
| tensor_size | 输入 | Tensor数据的长度。单位为字节。 |
| total_size | 输出 | 创建出的Tensor在内存中的长度。和tensor_size参数不同,total_size包含Tensor对象的长度和Tensor数据的长度。单位为字节。 |