![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch中的函数
piex_l
上海某学校学生
展开
-
AttributeError: ‘DataParallel‘ object has no attribute ‘save‘
膜拜大神原创 2021-12-13 22:03:47 · 4275 阅读 · 0 评论 -
2021-10-28
PyTorch提供了几种常见的参数初始化方式的实现Xavier Initialization: 基本思想是维持输入和输出的方差一致,避免了所有的输出值都为0, 使用于任何激活函数Xavier 均匀分布:torch.nn.init.xavier_uniform_(tensor, gain = 1),服从均匀分布U(-a, a), # 分布参数a=gain * sqrt(6 / (fan_in + fan_out)),gain的大小由激活函数的类型来决定。 # 其中fan_in是指第i层神原创 2021-10-28 19:51:58 · 105 阅读 · 0 评论 -
2021-10-25
pytorch中的两种转置函数的区别transpose()permute()transpose()torch.transpose(input, dim0, dim1, out=None) → Tensor函数返回输入矩阵input的转置。交换维度dim0和dim1参数:input (Tensor) – 输入张量,必填dim0 (int) – 转置的第一维,默认0,可选dim1 (int) – 转置的第二维,默认1,可选两个相关的交换的位置参数。permute()参数:dims原创 2021-10-25 12:40:51 · 339 阅读 · 0 评论