When I set batch size to 512, an error is reported：
RuntimeError: nonzero is not supported for tensors with more than INT_MAX elements, file a support request
But when I set batch size to 256， it will run， why？
Report error code at ：
def dense_to_sparse(tensor):
    r"""Converts a dense adjacency matrix to a sparse adjacency matrix defined
    by edge indices and edge attributes.

    Args:
        tensor (Tensor): The dense adjacency matrix.
     :rtype: (:class:`LongTensor`, :class:`Tensor`)
    """
    assert tensor.dim() == 2
    index = tensor.nonzero(as_tuple=False).t().contiguous()
    value = tensor[index[0], index[1]]
    return index, value

Thank you very much for your reply