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

PyTorch学习笔记之Tensors 2

时间:2017-07-21 22:01:15      阅读:209      评论:0      收藏:0      [点我收藏+]

标签:future   bsp   nbsp   初始化   学习   class   imp   copy   out   

Tensors的一些应用

 1 ‘‘‘
 2 Tensors和numpy中的ndarrays较为相似, 因此Tensor也能够使用GPU来加速运算
 3 ‘‘‘
 4 # from _future_ import print_function
 5 import torch
 6 x = torch.Tensor(5, 3)        # 构造一个未初始化的5*3的矩阵
 7 
 8 x2 = torch.rand(5, 3)         # 构造一个随机初始化的矩阵 the same as
 9 
10 # print(x.size())              # torch.Size([5, 3])
11 y = torch.rand(5, 3)
12 
13 # x+y 
14 # method 1
15 print(torch.add(x, y))
16 # 输出tensor
17 result = torch.Tensor(5, 3)
18 
19 # method 2
20 print(torch.add(x, y, out=result))    # the same as the above one
21 # print(result)                   result=torch.add(x, y, out=result)
22 
23 # method 3
24 print(y.add_(x))                 # 任何可以改变tensor内容的操作都会在方法名后加一个下划线‘_‘,例如:x.copy_(y), x.t_(), 这俩都会改变x的值。
25 # print(y)                       y=y.add_(x)
26 
27 # print(x[:, 1])                        # 这一操作会输出x矩阵的第二列的所有值

 

PyTorch学习笔记之Tensors 2

标签:future   bsp   nbsp   初始化   学习   class   imp   copy   out   

原文地址:http://www.cnblogs.com/Joyce-song94/p/7219661.html

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