Function dtype_of

python/dgl/utils/internal.py:1171–1175

Return the dtype of the data which can be either a tensor or a dict of tensors.

(data)

def dtype_of(data):
    """Return the dtype of the data which can be either a tensor or a dict of tensors."""
    return F.dtype(
        next(iter(data.values())) if isinstance(data, Mapping) else data
    )


_init_api("dgl.utils.internal")