[关闭]
@devilloser 2018-12-27T12:14:42.000000Z 字数 598 阅读 770

pytorch tensor api

pytorch


  1. a=torch.rand(3,4)
  2. b=torch.rand(3,4)
  3. a.ceil(input,out=None)#进一法
  4. a.clamp(min,max)#裁剪
  5. a.expand(*size)#复制
  6. a.gather(dim,index)#取出一个和index一个shapetensor,在dim维取
  7. a.index_add_(dim,index,tensor)#在dim维特定index上加上tensor
  8. a.index_copy_(dim,index,tensor)
  9. a.index_fill_(dim,index,val)
  10. a.index_select(dim,index)
  11. torch.lerp(start,end,weight)#outi=starti++weight(endi-starti)
  12. torch.logsumexp(input,dim,keepdim=False,out=None)
  13. a.map_(tensor,callable)#atensor element-wise 执行callable
  14. a.masked_scatter(mask,b)#在mask为1的位置讲b copya
  15. a.masked_select(mask)#返回mask为1的所有数值
  16. a.permute(*dims)#permute the dimensions of this tensor
添加新批注
在作者公开此批注前,只有你和作者可见。
回复批注