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):

将此对象转为指定类型

你可能感兴趣的:(torch.Storage常用函数)