美文网首页
torch.Storage常用函数

torch.Storage常用函数

作者: zhaoQiang012 | 来源:发表于2018-08-01 16:51 被阅读0次

    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常用函数

          本文链接:https://www.haomeiwen.com/subject/oryrvftx.html