About function batched_semi_loss()

[Humputer]

Thank you for your efforts! When I utilize this model to process a large graph, I've observed that the space complexity of the batched loss becomes O(N^2). Consequently, the GPU memory usage steadily increases while the loop 'for i in range(num_batches):' is executed. Looking forward to your reply!