DictToArray#
- class pinnx.nn.DictToArray(axis=-1, **units)[source]#
DictToArray layer, scaling the input data according to the given units, and merging them into an array.
- Parameters:
axis (
int) – The axis to concatenate.**units – The units for each input. The unit should be the instance of
brainunit.Unit, but it can be None.