torch.Storage常用函数

torch.Storage

本笔记引用自PyTorch中文文档

  • 一个torch.Storage是一个单一数据类型的连续一维数组
  • 每个torch.Tensor都有一个对应的,相同数据类型的存储
    class torch.FloatStorage

byte():

将此存储转为byte类型

char():

将此存储转为char类型

clone():

返回此存储的一个副本

cpu():

若当前此存储不在CPU上,返回一个它的CPU副本

cuda(device=None, async=False):

返回此对象在CUDA内存中的一个副本

type(new_type=None, async=False):

将此对象转为指定类型

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容