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()代替