Public Member Functions
	UnitStatFeaturizer ()

Public Member Functions inherited from cherrypi::UnitAttributeFeaturizer
virtual	~UnitAttributeFeaturizer ()=default

virtual Data	extract (State *state, UnitsInfo::Units const &units, Rect const &boundingBox=Rect())
	Extract unit features for a given set of units. More...

Data	extract (State *state, Rect const &boundingBox=Rect())
	Extract unit features for all live units. More...

Data	extract (State *state, UnitFilter filter, Rect const &boundingBox=Rect())
	Extract unit features for all live units that pass the given filter. More...

FeatureData	toSpatialFeature (Data const &data, SubsampleMethod pooling=SubsampleMethod::Sum) const
	Embeds the unit attribute data into a spatial feature. More...

void	toSpatialFeature (FeatureData *dest, Data const &data, SubsampleMethod pooling=SubsampleMethod::Sum) const
	Embeds the unit attribute data into a spatial feature. More...

Protected Member Functions
virtual void	extractUnit (TensorDest, cherrypi::Unit *) override
	Reimplement this in actual featurizers. More...

Additional Inherited Members
Public Types inherited from cherrypi::UnitAttributeFeaturizer
using	UnitFilter = std::function< bool(Unit *)>

using	TensorDest = torch::TensorAccessor< float, 1 >

Public Attributes inherited from cherrypi::UnitAttributeFeaturizer
std::shared_ptr< BaseJitter >	jitter = std::make_shared<NoJitter>()
	Optional jittering of unit positions. More...

CustomFeatureType	type

std::string	name

int	numChannels

Constructor & Destructor Documentation

cherrypi::UnitStatFeaturizer::UnitStatFeaturizer ( )

inline

Member Function Documentation

void cherrypi::UnitStatFeaturizer::extractUnit	(	TensorDest	acc,
		cherrypi::Unit *	unit
	)

overrideprotectedvirtual

Reimplement this in actual featurizers.

This function is expected to set acc[0], ..., acc[numChannels-1]

int constexpr cherrypi::UnitStatFeaturizer::kNumChannels

static

Initial value:

=

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