码迷,mamicode.com
首页 > 其他好文 > 详细

pytorch入门与实践-2.2

时间:2019-02-26 09:12:08      阅读:169      评论:0      收藏:0      [点我收藏+]

标签:pytorch   rand   支持   ORC   uda   数组   维度   实践   大小   

Tensor

1--本质上可以理解为具有不同维度的数组

2--支持的基本运算

|---创建Tensor:  x=t.tensor(x,y) x,y 表示数组的大小 , x=t.rand(x,y), x=t.ones(x,y)

|---获取Tensor的大小 x.size() 

|---加法运算

|----普通加法(返回新的变量) 

  x=t.tensor(a,b),y=t.tensor(a,b),

  x+y

  t.add(x,y)

  x.add(y)

  r=t.tensor(a,b), t.add(x,y,out=r)

|----改变运算对象的加法

  y.add_(x)   y的内容被改变了

3--与numpy的关系

  本质上就是共享一个内存的对象的,可以快速转化

|----tensor->numpy

  x=t.tensor(a,b),y=x.numpy()

|----numpy->tensor

  x=np.ones(a,b),y=t.from_numpy(x)

4--与cuda的tensor的关系可以通过  x.cuda()转化 

  

pytorch入门与实践-2.2

标签:pytorch   rand   支持   ORC   uda   数组   维度   实践   大小   

原文地址:https://www.cnblogs.com/CXianRen/p/10434905.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!