Skip to content
Snippets Groups Projects
Commit ef539ba7 authored by Thibault Allenet's avatar Thibault Allenet
Browse files

Add tensor constructor from dimensions

parent bed0f107
No related branches found
No related tags found
No related merge requests found
...@@ -164,6 +164,20 @@ class Tensor : public Data, ...@@ -164,6 +164,20 @@ class Tensor : public Data,
{ {
// ctor // ctor
} }
/**
* @brief Construct a new Tensor object from dimensions.
*
* @param dims dimensions of the tensor
* @param dataType datatype of the tensor (default = DataType::Float32)
*/
Tensor(std::vector<DimSize_t> dims, DataType dataType = DataType::Float32)
: Data(Type),
mDataType(dataType),
mDims(dims)
{
computeSize();
}
/** /**
* @brief Construct a new Tensor object copied from another one. * @brief Construct a new Tensor object copied from another one.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment