博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
pytorch torch.Stroage();torch.cuda()
阅读量:4686 次
发布时间:2019-06-09

本文共 1084 字,大约阅读时间需要 3 分钟。

转自:

 

torch.Storage是单个数据类型的连续的一维数组,每个torch.Tensor都具有相同数据类型的相应存储。他是torch.tensor底层数据结构,他除了像Tensor一样定义数值,还可以直接把文件映射到内存中进行操作,如果你使用的是pytorch神经网络,你不需要直接使用它们。中文文档地址:

 

注意:任何比一维数组更复杂的都需要用到张量。

torch.Storage主要有以下几种类型:

torch.FloatStorage([1,2,3,4,5])torch.ByteStorage([1,2,3,4,5])torch.ShortStorage([1,2,3,4,5])torch.IntStorage([1,2,3,4,5])torch.LongStorage([1,2,3,4,5])torch.FloatStorage([1,2,3,4,5])torch.DoubleStorage([1,2,3,4,5])torch.ShortStorage([1,2,3,4,5])

一、构造函数

  • 你可以使用torch.FloatStorage()构造一个空的Storage,构造函数的数值是随机初始化的。
  • 我们可以使用2种方法来构造torch.FloatStorage()并把它存入内存中
# 通过数组torch.FloatStorage([1,2,3,4,5])# 随机生成一个Storagetorch.FloatStorage(5)

二、torch.Storage的操作以及类型的转变

  • torch.Storage包含多种操作例如:size()返回此存储转的大小,tolist()返回一个包含此存储中元素的列表等等,具体参考文档
  • torch.FloatStorage()类型之间的转换,其实文档里有介绍,我这里给个例子:
a = torch.FloatTensor([1,2,3,4])print a.char()print a.long()

三、torch.Storagetorch.Tensor之间的转化

将Tensor转换为Storage非常简单,只需要storage()函数即可

import torcha = torch.FloatTensor([1,2,3,4])print a.storage()

它没有其他高级的用法,它是张量的一层。它可以共享多个张量的相同存储。或者更有效地处理一些操作。

 

 

torch.cuda()

转自:

 

 

 

 

 

 

 

 

 

 

 

 

 

1

转载于:https://www.cnblogs.com/Manuel/p/10997042.html

你可能感兴趣的文章
maven3在eclipse3.4.2中创建java web项目
查看>>
发布时间 sql语句
查看>>
黑马程序员 ExecuteReader执行查询
查看>>
记一些从数学和程序设计中体会到的思想
查看>>
题目1462:两船载物问题
查看>>
POJ 2378 Tree Cutting(树形DP,水)
查看>>
第二冲刺阶段个人博客5
查看>>
UVA 116 Unidirectional TSP (白书dp)
查看>>
第三方测速工具
查看>>
MySQL 网络访问连接
查看>>
在aws ec2上使用root用户登录
查看>>
数据访问 投票习题
查看>>
CIO知识储备
查看>>
cnblog!i'm coming!
查看>>
使用点符号代替溢出的文本
查看>>
Axios 中文说明
查看>>
fatal: remote origin already exists.
查看>>
gridview 自定义value值
查看>>
2018二月实现计划成果及其三月规划
查看>>
封装springmvc处理ajax请求结果
查看>>