参数¶

• input (Tensor) – 维度为 \([N_1, N_2, ..., N_k, D]\) 的多维Tensor，其中最后一维D是类别数目。数据类型为float32或float64。

• label (Tensor) – 输入input对应的标签值。若soft_label=False，要求label维度为 \([N_1, N_2, ..., N_k]\) 或 \([N_1, N_2, ..., N_k, 1]\) ，数据类型为int64，且值必须大于等于0且小于D；若soft_label=True，要求label的维度、数据类型与input相同，且每个样本各软标签的总和为1。

• weight (Tensor, optional) – 权重张量，需要手动给每个类调整权重，形状是（C）。它的维度与类别相同，数据类型为float32，float64。默认值为None。

• ignore_index (int) – 指定一个忽略的标签值，此标签值不参与计算，负值表示无需忽略任何标签值。仅在soft_label=False时有效。 默认值为-100。

• reduction (str, optional) – 指示如何按批次大小平均损失，可选值为"none","mean","sum"，如果选择是"mean"，则返回减少后的平均损失；如果选择是"sum"，则返回减少后的总损失。如果选择是"none"，则返回没有减少的损失。默认值是“mean”。

• soft_label (bool, optional) – 指明label是否为软标签。默认为False，表示label为硬标签；若soft_label=True则表示软标签。

• axis (int, optional) - 进行softmax计算的维度索引。 它应该在 \([-1，dim-1]\) 范围内，而 dim 是输入logits的维度。 默认值：-1。

• name (str，optional） - 操作的名称(可选，默认值为None）。更多信息请参见 Name。

返回¶

表示交叉熵结果的Tensor，数据类型与input相同。若soft_label=False，则返回值维度与label维度相同；若soft_label=True，则返回值维度为 \([N_1, N_2, ..., N_k, 1]\) 。

代码示例¶

import paddle

input_data = paddle.rand(shape=[5, 100])
label_data = paddle.randint(0, 100, shape=[5,1], dtype="int64")
weight_data = paddle.rand([100])

loss = paddle.nn.functional.cross_entropy(input=input_data, label=label_data, weight=weight_data)
print(loss)
# [4.38418674]