Is there an equivalent for edata and ndata that tracks the graph level data? Perhaps a gdata?
Yes it is possible to just track this separate from the underlying graphs it’s own tensor. However, that adds a lot of overhead to the simple batch and unbatch ops.