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 element quantity and data type.

    static std::unique_ptr<uint8_t[]> CreateFollowing(const int64_t shape_size, const ge::DataType dt, size_t &total_size)

  • 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

Table 1 Parameters (passing element quantity and data type for Tensor creation)

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.

Table 2 Parameters (passing data type and tensor length for Tensor creation)

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