pytorch
咖啡味儿的咖啡
这个作者很懒,什么都没留下…
展开
-
【pytorch】 pytorch nn.init 中实现的初始化函数
https://pytorch.org/docs/stable/nn.html#torch-nn-init 1. 均匀分布 torch.nn.init.uniform_(tensor, a=0, b=1) 服从~U(a,b) U(a, b)U(a,b) 2. 正太分布 torch.nn.init.normal_(tensor, mean=0, std=1) 服从~N(mean,std) N...转载 2019-03-29 16:59:43 · 4605 阅读 · 0 评论 -
【PyTorch】Tensor_把索引标签转换成one-hot标签表示
对于分类问题,标签可以是类别索引值也可以是one-hot表示。以10类别分类为例,lable=[3] 和label=[0, 0, 0, 1, 0, 0, 0, 0, 0, 0]是一致的. 现在给定索引标签,怎么将其转换为one-hot标签表示? >>>class_num = 10 >>>batch_size = 4 >>>label = t...转载 2019-04-28 13:47:57 · 2998 阅读 · 0 评论