pytorch
X-ocean
这个作者很懒,什么都没留下…
展开
-
pytorch:项目结构
见到了一个非常elegant的项目代码结构,来自。原创 2023-07-14 17:40:00 · 86 阅读 · 0 评论 -
Pytorch笔记:关于MSELoss的用法
MSELoss的reduction参数有三个取值,分别是mean, sum和none,一直搞不太清楚,所以这里写个笔记记录一下。1. mean当reduction参数设置为mean时,会返回一个shape为[]的标量,其值是每个位置上元素的差的平方的和的均值。输出:2. sum当reduction参数设置为sum时,会返回一个shape为[]的标量,其值是每个位置上元素的差的平方的和的总和。输出:3. none当reduction参数设置为none时,保留原始维度原创 2023-04-09 23:12:59 · 678 阅读 · 0 评论 -
Pytorch笔记:张量操作
记录一些关于张量的操作,会持续更新。原创 2022-12-30 09:29:28 · 466 阅读 · 0 评论