torch.Storage常用函数

2018-08-01  本文已影响0人  zhaoQiang012

torch.Storage

本笔记引用自PyTorch中文文档

byte():

将此存储转为byte类型

char():

将此存储转为char类型

clone():

返回此存储的一个副本

cpu():

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

cuda(device=None, async=False):

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

type(new_type=None, async=False):

将此对象转为指定类型

上一篇下一篇

猜你喜欢

热点阅读