动手学深度学习笔记
动手学深度学习笔记
Nice25
这个作者很懒,什么都没留下…
展开
-
nn.functional.softmax(X, dim=-1)
dim表示在该维度应用Softmax函数,将值标准化为概率分布。原创 2024-10-09 21:17:07 · 116 阅读 · 0 评论 -
PyTorch中None的使用
None 的使用允许我们在进行比较时,确保每个张量的形状正确,以便利用广播功能进行逐元素的比较操作。通过插入新维度,我们可以很方便地处理不同形状的张量,而不需要手动扩展或改变它们的大小。原创 2024-10-09 21:11:44 · 219 阅读 · 0 评论 -
PyTorch 中 reshape 函数用法示例
reshape 是用于改变张量形状的工具,数据不变。可以使用 -1 进行自动推导。适用于多维张量的重塑,便于后续的数据处理和建模。原创 2024-10-09 20:58:25 · 179 阅读 · 0 评论