首页 > 代码库 > PyTorch学习笔记之Tensors 2
PyTorch学习笔记之Tensors 2
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
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。