Navigation
API > API/Runtime > API/Runtime/NNE
References
| Module | NNE |
| Header | /Engine/Source/Runtime/NNE/Public/NNETypes.h |
| Include | #include "NNETypes.h" |
Syntax
class FTensorDesc
Remarks
The descriptor for a tensor as model inputs and output.
A tensor is described by its name, the type of data it contains and it's shape. Since input and output tensors of a neural network can have dynamic shapes, Shape is symbolic.
Constructors
No constructors are accessible with public or protected access.
Functions
| Type | Name | Description | |
|---|---|---|---|
| ENNETensorDataType | GetDataType () |
Get the data type of the tensor | |
| uint32 | Get the size in bytes of one element of the tensor. | ||
| const FString & | GetName () |
Get the name of the tensor | |
| const FSymbolicTensorShape & | GetShape () |
Get the symbolic shape of the tensor. | |
| FTensorDesc | Make
(
const FString& Name, |
Construct a tensor description for model input/output |