dgl.BatchedDGLGraph.batch_sizeΒΆ