class SoftmaxLoss(embedding_size: int, num_groups: int, temperature: float = 0.05)[源代码]¶ 基类：GroupLoss 常规交叉熵损失。 Softmax 点积的一种实现。它旨在与基类 GroupLoss 配合使用。 参数: embedding_size – 编码器的输出维度。 num_groups – 数据集中的组数量。 temperature – 用于除 logits 的温度值，默认为 0.05 forward(embeddings: Tensor, groups: LongTensor) → Tensor[源代码]¶ 计算损失值。 embeddings – 形状：(batch_size, vector_length) - 来自编码器的输出嵌入 embedding_size – 编码器的输出维度。 groups – 形状：(batch_size,) - 与嵌入相关的组 ID 返回值: Tensor – 零尺寸张量，损失值 training: bool¶ 下一页 上一页
