pytorch常用函数:torch.nn.functional.binary_cross_entropy_with_logits

torch.nn.functional.binary_cross_entropy_with_logits

在这里插入图片描述

它的参数有:
input, target,weight,size_average,reduce,reduction,pos_weight,下面分别说一下每个参数的用法

input

input–任意形状的张量目标–与输入形状相同的张量

target

target:与输入形状相同的张量

weight

weight(Tensor,可选)–手动调整比例的权重(如果重复进行以匹配输入张量形状)

size_average

size_average(bool,可选)–已弃用(请参阅reduction)。默认情况下,损失是批次中每个损失元素的平均数。请注意,对于某些损失,每个样本有多个元素。如果该字段size_average 设置为False,则每个小批量的损失总和。当reduce为时被忽略False。默认:True

reduce

reduce(bool,可选)–不推荐使用(请参阅reduction)。默认情况下,根据,对每个小批量的损失进行平均或求和size_average。当reduce为is时False,则返回每批元素的损失,然后忽略size_average。默认:True

reduction

reduction(字符串,可选)–指定要应用于输出的减少量: ‘none’| ‘mean’| ‘sum’。‘none’:不应用缩减, ‘mean’:输出的总和除以输出中元素的数量,‘sum’:输出的总和。注意:size_average 和reduce正在弃用的过程中,同时,指定这两个args中的任何一个将覆盖reduction。默认:‘mean’

pos_weight

pos_weight(Tensor,可选)–正例的权重。必须是长度等于类数的向量。

例子

>>> input = torch.randn(3, requires_grad=True)
>>> target = torch.empty(3).random_(2)
>>> loss = F.binary_cross_entropy_with_logits(input, target)
>>> loss.backward()
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值