Navigation
API > API/Plugins > API/Plugins/NNERuntimeBasicCpu > API/Plugins/NNERuntimeBasicCpu/FModelBuilder
Description
Make a layer which aggregates a set of other observations using attention. This is used by LearningAgents.
| Name | MakeAggregateSet |
| Type | function |
| Header File | /Engine/Plugins/Experimental/NNERuntimeBasicCpu/Source/NNERuntimeBasicCpu/Public/NNERuntimeBasicCpuBuilder.h |
| Include Path | #include "NNERuntimeBasicCpuBuilder.h" |
| Source | /Engine/Plugins/Experimental/NNERuntimeBasicCpu/Source/NNERuntimeBasicCpu/Private/NNERuntimeBasicCpuModel.cpp |
FModelBuilderElement MakeAggregateSet
(
const uint32 MaxElementNum,
const uint32 OutputEncodingSize,
const uint32 AttentionEncodingSize,
const uint32 AttentionHeadNum,
const FModelBuilderElement & SubLayer,
const FModelBuilderElement & QueryLayer,
const FModelBuilderElement & KeyLayer,
const FModelBuilderElement & ValueLayer
)