torch.Storage
本笔记引用自PyTorch中文文档
- 一个
torch.Storage是一个单一数据类型的连续一维数组 - 每个
torch.Tensor都有一个对应的，相同数据类型的存储
class torch.FloatStorage
byte():
将此存储转为byte类型
char():
将此存储转为char类型
clone():
返回此存储的一个副本
cpu():
若当前此存储不在CPU上，返回一个它的CPU副本
cuda(device=None, async=False):
返回此对象在CUDA内存中的一个副本
type(new_type=None, async=False):
将此对象转为指定类型
网友评论