pytorch loss function 总结

最近看了下 PyTorch 的损失函数文档,整理了下本身的理解,从新格式化了公式以下,以便之后查阅。html 值得注意的是,不少的 loss 函数都有 size_average 和 reduce 两个布尔类型的参数,须要解释一下。由于通常损失函数都是直接计算 batch 的数据,所以返回的 loss 结果都是维度为 (batch_size, ) 的向量。python 若是 reduce = Fal
相关文章
相关标签/搜索