栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

day24 Storage 操作1 入门 一起学 PyTorch吧,PyTorch从入门开始 每天都更新 超详细 参数 常用方法 简单案列 共同学习,共同进步 坚持

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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 中——或正好相反,则复制操作相对于宿主异步执行。否则此参数不起效果。
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/714392.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号