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里面都有,不过是把函数调用变成方法调用。

除此之外,这个方法里面还多了一个就地操作,就是直接修改原始对象而不创建新的对象,这意味着操作的结果会直接映射到原始对象上,而不需要分配额外的内存。这个库就这些内容。

Share

发表回复

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

Post comment

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