torch.Storage

---

torch.Storage是单个数据类型的连续的一维数组，每个torch.Tensor都具有相同数据类型的相应存储。

class torch.FloatStorage 

• byte()：将 Storage 转为 byte 类型
• char():将 Storage 转为 char 类型
• clone():返回 Storage 的副本
• copy_()
• cpu()：如果尚未在 CPU 上，则返回 Storage 的 CPU 副本

• cuda(device=None, async=False)：在 CUDA 内存中返回此对象的副本。如果该对象已经在 CUDA 内存中，并且在正确的设备上，则不会执行任何操作，并返回原始对象。 参数说明：

  1. device (int) - 目标 GPU 的 id。默认值是当前设备。
  2. async (bool) -如果值为 True，且源在锁定内存中，则副本相对于宿主是异步的。否则此参数不起效果。
  3. data_ptr()： 返回一个时间戳
  4. double()：将 Storage 转为 double 类型
  5. element_size()：返回参数的 size
  6. fill_()
  7. float():将 Storage 转为 float 类型
  8. from_buffer()
  9. half():将 Storage 转为 half 类型
  10. int():将 Storage 转为 int 类型
  11. is_cuda = False
  12. is_pinned()
  13. is_shared()
  14. is_sparse = False
  15. long():将 Storage 转为 long 类型
  16. new()
  17. pin_memory()：将存储复制到固定内存（如果尚未固定）。
  18. resize_()
  19. share_memory_()：这对于已经在共享内存和 CUDA 存储器中的存储器是无效的，不需要为了跨进程共享而移动。无法调整共享内存中的存储空间。返回：self
  20. short():将 Storage 转为 short 类型
  21. size()：返回 Storage 转的大小
  22. tolist():返回一个包含 Storage 中元素的列表
  23. type(new_type=None, async=False)：将此对象转为指定类型。如果已经是正确类型，不会执行复制操作，直接返回原对象。

参数说明：

1. new_type (type 或 string) -需要转成的类型
2. async (bool) -如果值为 True，并且源处于固定内存中，目标位于 GPU 上，反之亦然，则相对于主机异步执行该副本。否则，参数没有效果。

具体使用教程请参考：使用 torch.Storage 共享多个张量的相同存储以及将 torch.Tensor 转化为 torch.Storage

译者署名

用户名	头像	职能	签名
Song		翻译	人生总要追求点什么

---

我们一直在努力

apachecn/AiLearning

---