摘要: softmax中dim维度这个参数的取值 准则是dim等于多少就是,该索引维度数据可以变化,dim=0, 二维 索引 [0,0]+[1,0]+[2,0],三维度 [0,0,0]+[1,0,0]+[2,0,0],dim=1,[0,0]+[0,1]+[0,2] 代码: import torch impo 阅读全文
posted @ 2021-03-25 22:18 司砚章 阅读(49) 评论(0) 推荐(0) 编辑
摘要: 构造线性模型 code: import torch import numpy as np import torchvision #torch的视觉包 import torchvision.datasets as datasets import torchvision.transforms as tr 阅读全文
posted @ 2021-03-25 11:35 司砚章 阅读(42) 评论(0) 推荐(0) 编辑