PyTorch中的张量(Tensor)数据类型主要包括以下几种:
这些数据类型可以通过torch.Tensor
类的dtype
属性进行设置和查询。例如,要创建一个32位浮点数的张量,可以使用以下代码:
import torch
x = torch.Tensor([1.0, 2.0, 3.0], dtype=torch.float32)
print(x)
此外,PyTorch还支持自定义数据类型,通过继承torch.Tensor
类并重写__new__
方法来实现。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
推荐阅读:keyvaluepair数据类型有哪些