torch.softmax()
torch.reduce_max()
torch.pow()
torch.subtrack()=torch.sub()
torch.div()=torch.divide()
torch.gather()
torch.log()
pytorch常用计算
pytorch学习笔记——常用函数
于 2022-11-29 11:49:09 首次发布
这篇博客介绍了PyTorch中几个关键的运算函数,包括torch.softmax用于激活函数的平滑处理,torch.reduce_max用于查找张量中的最大值,torch.pow进行指数运算,torch.subtract和torch.div进行减法和除法操作,以及torch.gather用于按索引选取数据。这些函数在深度学习模型的构建和训练中起着重要作用。


被折叠的 条评论
为什么被折叠?



