logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

Pytorch Softmax用法

Pytorch Softmax用法pytorch中的softmax主要存在于两个包中分别是:torch.nn.Softmax(dim=None)torch.nn.functional.softmax(input, dim=None, _stacklevel=3, dtype=None)下面分别介绍其用法:torch.nn.Softmaxtorch.nn.Softmax中只要一个参数:来制定归一化维

文章图片
#pytorch#深度学习#神经网络
到底了