对pytorch中softmax(dim=x)的理解

先看看文档怎么说的



这里说指定某个纬度（dim)，就会让那个纬度的和为1

看一下input是2维的情况

dim=0

dim=1

再看看input是3维的情况

dim = 0


dim = 1

pytorch

相关

pytorch网络转libtorch常见问题

pytorch 中的常见问题汇总

关于pytorch中损失函数的参数说明

pytorch的显存释放机制torch.cuda.empty_cache()

Pytorch学习-多层感知机

OpenKE-PyTorch使用

Pytorch训练时显存分配过程探究

Pytorch的view方法

pytorch之torch.nn.Conv2d()函数详解

Pytorch CrossEntropyLoss 用例

PyTorch项目使用TensorboardX进行训练可视化

Pytorch模型保存与加载，并在加载的模型基础上继续训练

标签