上传者: 38522029
|
上传时间: 2022-04-16 03:02:08
|
文件大小: 123KB
|
文件类型: PDF
PyTorch | (1)初识PyTorch
PyTorch | (2)PyTorch 入门-张量
PyTorch | (3)Tensor及其基本操作
Tensor attributes:
在tensor attributes中有三个类,分别为torch.dtype, torch.device, 和 torch.layout
其中, torch.dtype 是展示 torch.Tensor 数据类型的类,pytorch 有八个不同的数据类型,下表是完整的 dtype 列表.
Torch.device 是表现 torch.Tensor被分配的设备类型的类,其中分为’cpu’ 和 ‘cuda’