CreateFollowing
Description
Creates a tensor with the specified data type and size, whose data is continuously arranged after the tensor object.
Prototype
- Creates a tensor based on the passed data type and tensor length.
static std::unique_ptr<uint8_t[]> CreateFollowing(const ge::DataType dt, const size_t tensor_size, size_t &total_size)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
shape_size |
Input |
Number of elements. |
dt |
Input |
Data type. The value is of the DataType type. |
total_size |
Output |
Length of the created tensor in the memory. Length of the tensor object and tensor data. |
Parameter |
Input/Output |
Description |
|---|---|---|
dt |
Input |
Data type. The value is of the DataType type. |
tensor_size |
Input |
Length of the tensor data, in bytes. |
total_size |
Output |
Length of the created tensor in the memory, in bytes. Unlike tensor_size, total_size contains the length of the tensor object and tensor data. |
Returns
Pointer to the created tensor
Restrictions
None
Example
size_t total_size; auto tensor_holder = Tensor::CreateFollowing(shape_size, tensor_desc.GetDataType(), total_size);