彻底解决Torch[mat1 dim 1 must match mat2 dim 0]

如果是卷积之后的层

一定要先化为一个维度

exp:

x = x.view(x.size(),-1)

ps:

view函数的解释

把原先tensor中的数据按照行优先的顺序排成一个一维的数据（这里应该是因为要求地址是连续存储的），然后按照参数组合成其他维度的tensor。比如说是不管你原先的数据是[[[1,2,3],[4,5,6]]]还是[1,2,3,4,5,6]，因为它们排成一维向量都是6个元素，所以只要view后面的参数一致，得到的结果都是一样的。比如

a=torch.Tensor([[[1,2,3],[4,5,6]]])
b=torch.Tensor([1,2,3,4,5,6])print(a.view(1,6))
print(b.view(1,6))

得到的结果都是tensor([[1., 2., 3., 4., 5., 6.]])

等于是将不同维度的数据化为一个维度，因为全连接层是只能接受一维度的数据的

由图可见全连接层必须是一维数据：

同时全连接下一层的输入必须上一层的输出

exp:

self.fc1 = nn.Linear(64*12*4, 256)
self.fc2 = nn.Linear(256, 128)
self.fc3 = nn.Linear(128, 64)

彻底解决Torch[mat1 dim 1 must match mat2 dim 0]相关推荐

成功解决torch.cuda.CudaError: CUDA driver version is insufficient for CUDA runtime version (35) [ WARN:0
成功解决torch.cuda.CudaError: CUDA driver version is insufficient for CUDA runtime version (35) [ WARN:0 ...
成功解决torch\cuda\__init__.py“, line 208, in check_error raise Cuda Error(res) torch.cuda.Cuda Error: C
成功解决torch\cuda\__init__.py", line 208, in check_error raise Cuda Error(res) torch.cuda.Cuda Err ...
解决torch.cuda.is_available()为False的问题
解决torch.cuda.is_available为False 问题:在Anaconda环境下,电脑拥有GPU并且已经通过conda install安装了Pytorch.cudatoolkit,但是t ...
解决torch.cuda.is_available()返回False的问题
解决torch.cuda.is_available()返回False的问题 (1)可能是电脑不支持CUDA 我的电脑-右键-设备管理器-显示适配器.如果没有NVIDIAN开头的则电脑没有独立显卡,不能 ...
Pytorch安装教程及解决 torch.cuda.is_available() 返回 False 的问题
Pytorch安装教程及解决 torch.cuda.is_available() 返回 False 的问题有NVIDIA显卡的可以先去看CUDA安装教程,pytorch可以搭配CUDA10.0~ ...
Unity中训练一个ML-Agents项目—解决torch和mlagents配置问题
开始一个全新的Unity机器学习项目-ML-Agents 目录一.安装Anaconda并新建一个环境二.配置torch和mlagents 1. 进入虚拟环境(我的环境命名为ml-agents) 2 ...
亲测有效解决torch.cuda.is_available()返回False的问题（分析+多种方案），点进不亏
文章目录解决torch.cuda.is_available()返回False 出现返回False的原因问题1:版本不匹配问题2:错下成了cpu版本的(小编正是这种问题) 解决方案方案一方案二 ...
彻底解决兼容问题 Windows 7 安装 Visual C++ 6.0 (VC6) 全程图解
彻底解决兼容问题 Windows 7 安装 Visual C++ 6.0 (VC6) 全程图解参考文章: (1)彻底解决兼容问题 Windows 7 安装 Visual C++ 6.0 (VC6) ...
解决Missing artifact com.microsoft.sqlserver:sqljdbc4:jar:4.0问题
解决Missing artifact com.microsoft.sqlserver:sqljdbc4:jar:4.0问题参考文章: (1)解决Missing artifact com.micros ...

彻底解决Torch[mat1 dim 1 must match mat2 dim 0]

彻底解决Torch[mat1 dim 1 must match mat2 dim 0]相关推荐

最新文章

热门文章