pytorch相关

常用Tensor初始化

(基本的初始化)[https://pytorch.org/docs/stable/torch.html#creation-ops]
(与特定Tensor相同dtype和device的初始化，在自定义模型中使用可以不用分cpu和gpu了)[https://pytorch.org/docs/stable/tensors.html#torch.Tensor]
(说起来到底怎么判断一个模型属于cpu还是属于gpu……)

1	torch.eye(x.size(1))[x.argmax(1)]

先生成单位向量，然后用花式索引选择其对应行
很赞的方法

文章作者:Wuny

发布时间:2018-12-14, 19:04:27

最后更新:2019-02-01, 10:06:04

许可协议: "署名-非商用-相同方式共享 4.0" 转载请保留原文链接及作者。