Tensor中有属性torch.dtype(数据类型), torch.device(对象创建后存储在的设备名称) torch.layout(torch.Tensor内存布局的对象)三种属性
torch.Tensor([1,2,3],dtype=torch.float32,device=torch.device('cpu'))
torch.sparse_coo_tensor 定义稀疏的张量
默认cpu
使用cuda
三个参数
稀疏张量
稀疏转成稠密
增加2个参数
Tensor中有属性torch.dtype(数据类型), torch.device(对象创建后存储在的设备名称) torch.layout(torch.Tensor内存布局的对象)三种属性
torch.Tensor([1,2,3],dtype=torch.float32,device=torch.device('cpu'))
torch.sparse_coo_tensor 定义稀疏的张量
默认cpu
使用cuda
三个参数
稀疏张量
稀疏转成稠密
增加2个参数