Pytorch使用技巧（一）

1.pytorch使用GPU资源时需要显示调用，如果不指定的话，默认是使用cpu计算资源

input=input.to(device)

注意：一定要赋值于一个变量

2.torch随机矩阵：
 

a=torch.randn(1,3,3)
#生成1*3*3的随机矩阵

#tensorflow中的随机矩阵
b=tf.random_uniform([1,3,3])

3.在使用 torch.max时会返回两个矩阵，第一个是由某一维度最大值组成的矩阵，第二维度是由最大值所在的索引组成的矩阵。

4.torch中的expand_as(矩阵名)可通过tensorflow中的tf.tile（）代替