Navigation
API > API/Runtime > API/Runtime/NNE
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.
| Name | FTensorDesc |
| Type | class |
| Header File | /Engine/Source/Runtime/NNE/Public/NNETypes.h |
| Include Path | #include "NNETypes.h" |
Syntax
class FTensorDesc
Constructors
| Name | Remarks | Include Path | Unreal Specifiers |
|---|---|---|---|
FTensorDesc() |
NNETypes.h |
Variables
Protected
| Name | Type | Remarks | Include Path | Unreal Specifiers |
|---|---|---|---|---|
| DataType | ENNETensorDataType | NNETypes.h | ||
| Name | FString | NNETypes.h | ||
| Shape | FSymbolicTensorShape | NNETypes.h |
Functions
Public
| Name | Remarks | Include Path | Unreal Specifiers |
|---|---|---|---|
ENNETensorDataType GetDataType() |
Get the data type of the tensor | NNETypes.h | |
uint32 GetElementByteSize() |
Get the size in bytes of one element of the tensor. | NNETypes.h | |
const FString & GetName() |
Get the name of the tensor | NNETypes.h | |
const FSymbolicTensorShape & GetShape() |
Get the symbolic shape of the tensor. | NNETypes.h |
Static
| Name | Remarks | Include Path | Unreal Specifiers |
|---|---|---|---|
static FTensorDesc Make
(
const FString& Name, |
Construct a tensor description for model input/output | NNETypes.h |