PyTorch 1.0 中文文档:torch.Storage

译者:yuange250

torch.Storage 跟绝大部分基于连续存储的数据结构类似,本质上是一个单一数据类型的一维连续数组(array)。

每一个 torch.Tensor 都有一个与之相对应的torch.Storage对象,两者存储数据的数据类型(data type)保持一致。

下面以数据类型为float的torch.FloatStorage 为例介绍一下torch.Storage的成员函数。

class torch.FloatStorage
byte()

byte()函数可以将此storage对象的数据类型转换为byte

char()

char()函数可以将此storage对象的数据类型转换为char

clone()

clone()函数可以返回一个此storage对象的复制

copy_()
cpu()

阅读全文/改进本文

猜你喜欢

转载自www.cnblogs.com/wizardforcel/p/10392205.html