ArrayToDict#
- class pinnx.nn.ArrayToDict(axis=-1, **units)[source]#
Output layer, splitting the output data into a dict and assign the corresponding units.
- Parameters:
axis (
int) – The axis to split the output data.**units – The units of the output data. The unit should be the instance of
brainunit.Unit, but it can be None.