torch.Storage
本筆記引用自PyTorch中文文檔
- 一個(gè)
torch.Storage
是一個(gè)單一數(shù)據(jù)類型的連續(xù)一維數(shù)組 - 每個(gè)
torch.Tensor
都有一個(gè)對(duì)應(yīng)的哈踱,相同數(shù)據(jù)類型的存儲(chǔ)
class torch.FloatStorage
byte()
:
將此存儲(chǔ)轉(zhuǎn)為byte類型
char()
:
將此存儲(chǔ)轉(zhuǎn)為char類型
clone()
:
返回此存儲(chǔ)的一個(gè)副本
cpu()
:
若當(dāng)前此存儲(chǔ)不在CPU上睛蛛,返回一個(gè)它的CPU副本
cuda(device=None, async=False)
:
返回此對(duì)象在CUDA內(nèi)存中的一個(gè)副本
type(new_type=None, async=False)
:
將此對(duì)象轉(zhuǎn)為指定類型