UE::NNE::FTensorDesc Class Reference

#include <NNETypes.h>

Public Member Functions
const FString &	GetName () const
ENNETensorDataType	GetDataType () const
uint32	GetElementByteSize () const
const FSymbolicTensorShape &	GetShape () const

Static Public Member Functions
static NNE_API FTensorDesc	Make (const FString &Name, const FSymbolicTensorShape &Shape, ENNETensorDataType DataType)

Detailed Description

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.

Member Function Documentation

GetDataType()

ENNETensorDataType UE::NNE::FTensorDesc::GetDataType ( ) const

inline

Get the data type of the tensor

Returns

the data type of the tensor.

GetElementByteSize()

uint32 UE::NNE::FTensorDesc::GetElementByteSize ( ) const

inline

Get the size in bytes of one element of the tensor.

Returns

the size in bytes of one element of the tensor.

GetName()

const FString & UE::NNE::FTensorDesc::GetName ( ) const

inline

Get the name of the tensor

Returns

the name of the tensor.

GetShape()

const FSymbolicTensorShape & UE::NNE::FTensorDesc::GetShape ( ) const

inline

Get the symbolic shape of the tensor.

Returns

the symbolic shape of the tensor.

Make()

FTensorDesc UE::NNE::FTensorDesc::Make ( const FString &  Name, const FSymbolicTensorShape &  Shape, ENNETensorDataType  DataType  )

static

Construct a tensor description for model input/output

Parameters

Name	the name of the tensor.
Shape	the symbolic shape of the tensor.
DataType	the type of data the tensor contains.

Returns

a tensor shape with the given dimensions.

---

The documentation for this class was generated from the following files:

• Engine/Source/Runtime/NNE/Public/NNETypes.h
• Engine/Source/Runtime/NNE/Private/NNETypes.cpp