pytorch教程二:torch.Tensor
torch.Tensor
torch.Tensor是一种包含单一数据类型元素的多维矩阵,torch.Tensor定义了7种CPU tensor类型和8种GPU tensor 类型:
Data tyoe | CPU tensor | GPU tensor |
---|---|---|
32-bit floating point | torch.FloatTensor |
torch.cuda.FloatTensor |
64-bit floating point | torch.DoubleTensor |
torch.cuda.DoubleTensor |
16-bit floating point | N/A | torch.cuda.HalfTensor |
8-bit integer (unsigned) | torch.ByteTensor |
torch.cuda.ByteTensor |
8-bit integer (signed) | torch.CharTensor |
torch.cuda.CharTensor |
16-bit integer (signed) | torch.ShortTensor |
torch.cuda.ShortTensor |
32-bit integer (signed) | torch.IntTensor |
torch.cuda.IntTensor |
64-bit integer (signed) | torch.LongTensor |
torch.cuda.LongTensor |
torch.Tensor
是默认的tensor类型(torch.FlaotTensor
)的简称。
然后方法基本上在torch里面都有,不过是把函数调用变成方法调用。
除此之外,这个方法里面还多了一个就地操作,就是直接修改原始对象而不创建新的对象,这意味着操作的结果会直接映射到原始对象上,而不需要分配额外的内存。这个库就这些内容。