pytorch教程三:torch.Storage

torch.Storage

每一个torch.Tensor都有一个对应的、相同数据类型的存储。

byte()

将此存储转为byte类型

char()

将此存储转为char类型

clone()

返回此存储的一个副本

cpu()

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

cuda(device=None, async=False)

返回此对象在CUDA内存中的一个副本。
如果此对象已在CUDA内存中且在正确的设备上,那么不会执行复制操作,直接返回原对象。

参数:

  • device (int) – 目标GPU的id。默认值是当前设备。
  • async (bool) -如果值为True,且源在锁定内存中,则副本相对于宿主是异步的。否则此参数不起效果。

float()

将此存储转为float类型

double()

将此存储转为double类型

half()

将此存储转为half类型

int()

将此存储转为int类型

long()

将此存储转为long类型

short()

将此存储转为short类型

Share

1 Response

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

Post comment

 ©2025 [忧郁小王子的乌托邦] - 稳定运行: