下载数据集 import torch import torchvision import torchvision.transforms as transforms # 定义数据预处理的转换 transform = transforms.Compose( [transforms.ToTensor(), Read More
posted @ 2023-11-02 14:31 PythonNew_Mr.Wang Views(86) Comments(0) Diggs(0)
基本运算 import torch # 创建一个自定义的张量 t = torch.tensor([1.0, 2.0, 3.0]) # tensor([1., 2., 3.]) # 求平均值 t.mean() # tensor(2.) # 创建一个指定行列的张量 x = torch.empty(3,5 Read More
posted @ 2023-11-02 14:22 PythonNew_Mr.Wang Views(34) Comments(0) Diggs(0)