import torch


def one_hot_embedding(labels, num_classes):
    """
    将标签嵌入成one-hot形式
    :param labels: 标签,（LongTensor）类别标签,形状[N,]
    :param num_classes: 类别数,
    :return:(tensor)被编码的标签,形状（N,类别数）
    """
    # 返回2维张量，对角线全是1，其余全是0
    y = torch.eye(num_classes)
    return y[labels]