创建一个指定数据类型以及大小的Tensor,其数据在Tensor对象后连续排布。
static std::unique_ptr<uint8_t[]> CreateFollowing(int64_t shape_size, ge::DataType dt, size_t &total_size)
参数 |
输入/输出 |
说明 |
---|---|---|
shape_size |
输入 |
元素个数。 |
dt |
输入 |
数据类型,ge::DataType类型。 |
total_size |
输出 |
创建出的Tensor在内存中的长度。 |
创建的Tensor指针。
无。
size_t total_size; auto tensor_holder = Tensor::CreateFollowing(shape_size, tensor_desc.GetDataType(), total_size);