UserWarning: Implicit dimension choice for softmax has been deprecated

2024-01-03 12:32:11

文章目录

警告

解决

自 PyTorch 1.6 版本开始，F.softmax() 函数不再隐式选择维度，默认情况下不会对最后一个维度进行计算。为了解决这个警告并明确指定要应用 softmax 的维度，你需要修改代码如下：

# 假设 x 是一个形状为 (batch_size, num_classes) 或其他包含多个维度的张量
dim = -1  # 对于最后一维
prediction = F.softmax(x, dim=dim)

这里的 -1 表示最后一个维度，这是通常用于分类任务时 softmax 的应用位置。

文章来源:https://blog.csdn.net/qihshe/article/details/135349843
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。