pytorch教程三:torch.Storage
torch.Storage
每一个torch.Tensor都有一个对应的、相同数据类型的存储。
byte()
将此存储转为byte类型
char()
将此存储转为char类型
clone()
返回此存储的一个副本
cpu()
如果当前此存储不在CPU上,则返回一个它的CPU副本
cuda(device=None, async=False)
返回此对象在CUDA内存中的一个副本。
如果此对象已在CUDA内存中且在正确的设备上,那么不会执行复制操作,直接返回原对象。
参数:
float()
将此存储转为float类型
double()
将此存储转为double类型
half()
将此存储转为half类型
int()
将此存储转为int类型
long()
将此存储转为long类型
short()
将此存储转为short类型
1 Response