首页 > 其他 > 详细

PyTorch学习笔记之Tensors 2

时间:2017-07-21 21:29:24      阅读:283      评论:0      收藏:0      [点我收藏+]

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

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!