Skip to content

FNNEMultiDiscreteBuffer

Buffer for multi-discrete action or observation spaces.

Stores one selected value per multi-discrete dimension, where each dimension can have different numbers of possible values. Used when an agent makes several independent discrete decisions simultaneously.

struct FNNEMultiDiscreteBuffer : public FNNEPointBuffer

Methods

FNNEMultiDiscreteBuffer

FNNEMultiDiscreteBuffer() = default

FNNEMultiDiscreteBuffer

inline FNNEMultiDiscreteBuffer(int Size)

Constructor that initializes the multi-discrete buffer with a specific size.

Parameters

  • Size (int)

FNNEMultiDiscreteBuffer

virtual ~FNNEMultiDiscreteBuffer() = default

MakeBinding

inline UE::NNE::FTensorBindingCPU MakeBinding() const

Creates an NNE tensor binding for this buffer.


Accept

virtual void Accept(FNNEBufferVisitor &Visitor) const override

Accept method for the visitor pattern.

Parameters

  • Visitor (FNNEBufferVisitor)

Attributes

Buffer

TArray<int64> Buffer

Buffer storing the selected value for each multi-discrete dimension.

Source: Source/ScholaNNE/Public/NNEUtils/NNEBuffer.h