Blog Life

索引

1.改变张量的形状

1.CUDA，Tensor 使用.to把张量放置在任何设备上运行

1.torch数据类型

1.Returns the total number of elements in the input tensor. 返回元素数目 （torch.numel）

1.张量点乘， z = x.mul(y) ,维度相同 2.张量相乘， z = x.mm(y) , (m,n) x (n,p)

1.autograd自动微分 假如我们有一个向量x=(1,1)当成input，经过一系列运算得到了output变量y,如下图所示： 如图所示，向量x经过与4和自身相乘之后得到向量z，z再求长度，得到y 我们想要求y关于x的微分时，pytorch会帮我们自动求解。 from torch.autograd import Variable import torch x = Variable(torch...

1.output = input.clamp(min=arg1,max=arg2) 将input中的值放进min~max之间

知乎参考：https://zhuanlan.zhihu.com/p/36233589