摘要:
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 阅读全文
摘要:
构造线性模型 code: import torch import numpy as np import torchvision #torch的视觉包 import torchvision.datasets as datasets import torchvision.transforms as tr 阅读全文