![day24 Storage *** 作1 入门 一起学 PyTorch吧,PyTorch从入门开始 每天都更新 超详细 参数 常用方法 简单案列 共同学习,共同进步 坚持,第1张 day24 Storage *** 作1 入门 一起学 PyTorch吧,PyTorch从入门开始 每天都更新 超详细 参数 常用方法 简单案列 共同学习,共同进步 坚持,第1张](/aiimages/day24+Storage+%E6%93%8D%E4%BD%9C1+%E5%85%A5%E9%97%A8+%E4%B8%80%E8%B5%B7%E5%AD%A6+PyTorch%E5%90%A7%EF%BC%8CPyTorch%E4%BB%8E%E5%85%A5%E9%97%A8%E5%BC%80%E5%A7%8B+%E6%AF%8F%E5%A4%A9%E9%83%BD%E6%9B%B4%E6%96%B0+%E8%B6%85%E8%AF%A6%E7%BB%86+%E5%8F%82%E6%95%B0+%E5%B8%B8%E7%94%A8%E6%96%B9%E6%B3%95+%E7%AE%80%E5%8D%95%E6%A1%88%E5%88%97+%E5%85%B1%E5%90%8C%E5%AD%A6%E4%B9%A0%EF%BC%8C%E5%85%B1%E5%90%8C%E8%BF%9B%E6%AD%A5+%E5%9D%9A%E6%8C%81.png)
day24 Storage *** 作1 入门 一起学 PyTorch吧,PyTorch从入门开始 每天都更新 超详细 参数 常用方法 简单案列 共同学习,共同进步 坚持
一个
torch.Storage
是一个单一数据
类型的连续一维数组。每个
torch.Tensor
都有一个对应的、相同数据类型的存储。
class
torch
.
FloatStorage
byte()
:将此存储
转为
byte
类型
char()
:将此存储转为
char
类型
clone()
:返回此存储的一个副本
copy_():复制
cpu()
:如果当前此存储不在
CPU
上,则返回一个它的
CPU
副本
cuda(device=None, async=False)
返回此对象在
CUDA
内存中的一个副本。如果此对象已在
CUDA
内存中且在正确的设备上,那么不会执行复制 *** 作,直接返回原对象。
参数:
--
device (int) -
目标
GPU
的
id
。默认值是当前设备。
--
async (bool) -
如果值为
True
,且源在锁定内存中,则副本相对于宿主是异步的。 否则此参数不起效果。
double()
:将此存储转为
double
类型
float()
:将此存储转为
float
类型
half()
:将此存储转为
half
类型
int()
将此存储转为
int
类型
long():将此存储转为
long
类型
pin_memory()
如果此存储当前未被锁定,则将它复制到锁定内存中。
share_memory_() 将此存储移动到共享内存中。对于已经在共享内存中的存储或者CUDA 存储,这是一条空指令,
它们不需要移动就能在进程间共享。共享内存中的存
储不能改变大小。返回:
self
short()
将此存储转为
short
类型
tolist()
:返回一个包含此存储中元素的列表
type(new_type=None, async=False) 将此对象转为指定类型。如果已经是正确类型不会执行复制 *** 作,直接返回原对象。
参数:
--new_type (type or string) -
需要转成的类型
--
async (bool) -
如果值为
True
,且源在锁定内存中而目标在
GPU
中——或正好相反,则复制 *** 作相对于宿主异步执行。否则此参数不起效果。
评论列表(0条)