dokato

Thanks @jefferis , maybe it requires more thinking on design indeed. The idea was to have access within the `nlapply` loop to the attributes of the neuron. So for example...

Thanks, this indeed helps, or as I said I can clone it, but maybe such a feature in `pytorch_geometric` would be nice.