The title of your publication挨打

Page 95

第三章 torch.Storage 一个 torch.Storage 是一个单一数据类型的连续一维数组。 每个 torch.Tensor 都有一个对应的、相同数据类型的存储。 class torch.FloatStorage byte() 将此存储转为 byte 类型 char() 将此存储转为 char 类型 clone() 返回此存储的一个副本 copy_() cpu() 如果当前此存储不在 CPU 上,则返回一个它的 CPU 副本 cuda(device=None, async=False) 返回此对象在 CUDA 内存中的一个副本。 如果此对象已在 CUDA 内存中且在正确的设备上,那么不会执行复制操作,直接返回原对象。 参数:

 device (int) - 目标 GPU 的 id。默认值是当前设备。  async (bool) -如果值为 True,且源在锁定内存中,则副本相对于宿主是异步的。否则此参数不起 效果。 data_ptr() double() 将此存储转为 double 类型 element_size() fill_() float() 将此存储转为 float 类型 from_buffer() half() 将此存储转为 half 类型 int() 将此存储转为 int 类型 is_cuda = False is_pinned() is_shared() is_sparse = False long() 将此存储转为 long 类型 new() pin_memory() 如果此存储当前未被锁定,则将它复制到锁定内存中。 resize_()


Turn static files into dynamic content formats.

Create a flipbook
Issuu converts static files into: digital portfolios, online yearbooks, online catalogs, digital photo albums and more. Sign up and create your flipbook.