batch num nodes 和 batch num edges

[yzf-code]

请问是否可以在未来提供batch num nodes 和 batch num edges这两个函数呢？用来返回batch之后的大graph中的每个原来的小图各自有几个点几条边。类似dgl中的batch_num_nodes和batch_num_edges这样的function。