loss 加权_CrossEntropyLoss类别权重问题

官方文档写的是 weight的形状为【C】

eg:5个类的话 weight=tensor[1,1,1,1,1]

然后看了一下示例代码:

`import paddle

import numpy as np

input_data = paddle.uniform([5, 100], dtype="float64")

label_data = np.random.randint(0, 100, size=(5)).astype(np.int64)

weight_data = np.random.random([100]).astype("float64")

input = paddle.to_tensor(input_data)

label = paddle.to_tensor(label_data)

weight = paddle.to_tensor(weight_data)

ce_loss = paddle.nn.CrossEntropyLoss(weight=weight, reduction='mean')

output = ce_loss(input, label)

print(output)`

这里面,weight的shape是[100],但是这里不是只有5个类嘛。没有理解,请求帮助。谢谢!

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值