这是因为tf的文档要求label必须是一个单独的值，而不是一个one-hot编码后像[1,0,0], [0,1,0], [0,0,1].的值。
Can not squeeze dim[1], expected a dimension of 1, got 21
2021-11-17
495
版权
版权声明：
本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《
阿里云开发者社区用户服务协议》和
《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写
侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
简介：
Can not squeeze dim[1], expected a dimension of 1, got 21
目录
相关文章
|
8月前
|
机器学习/深度学习
PyTorch
算法框架/工具
归一化技术比较研究：Batch Norm, Layer Norm, Group Norm
本文将使用合成数据集对三种归一化技术进行比较，并在每种配置下分别训练模型。记录训练损失，并比较模型的性能。
446
2
2
|
9天前
|
TensorFlow
算法框架/工具
Python
|
1月前
|
机器学习/深度学习
PyTorch
算法框架/工具
详解三种常用标准化Batch Norm & Layer Norm & RMSNorm
通过本文的介绍，希望您能够深入理解Batch Norm、Layer Norm和RMSNorm的原理和实现，并在实际应用中灵活选择和使用，提升深度学习模型的性能和稳定性。
470
5
5
|
5月前
|
机器学习/深度学习
PyTorch
算法框架/工具
【Pytorch】Expected hidden[0] size (2, 136, 256), got [2, 256, 256]
文章解决了PyTorch中LSTM模型因输入数据的批次大小不一致导致的“Expected hidden[0] size”错误，并提供了两种解决方案：调整批次大小或在DataLoader中设置drop_last=True来丢弃最后一个不足批次大小的数据。
110
1
1
Expected more than 1 value per channel when training, got input size torch.Size
因为模型中用了batchnomolization，训练中用batch训练的时候当前batch恰好只含一个sample，而由于BatchNorm操作需要多于一个数据计算平均值，因此造成该错误。
961
0
0
|
PyTorch
算法框架/工具
pytorch报错 RuntimeError: The size of tensor a (25) must match the size of tensor b (50) at non-singleton dimension 1 怎么解决？
这个错误提示表明，在进行某个操作时，张量a和b在第1个非单例维（即除了1以外的维度）上的大小不一致。例如，如果a是一个形状为(5, 5)的张量，而b是一个形状为(5, 10)的张量，则在第二个维度上的大小不匹配。
4255
0
0
怎么将[tensor([[ 1, 2]]), tensor([[5, 6]]), tensor([[9, 10]])] 合并成 tensor([[1,2],[3,4],[5,6]])
可以先使用 torch.cat() 函数将列表中的张量在第0维（行）上进行拼接，然后再使用 .view() 函数将形状调整为需要的形状。
197
0
0
torch中如何将tensor([[1, 2, 3]]) 和 tensor([4]) 合并成 tensor([[1,2,3,4]])
可以使用 torch.cat() 方法将两个张量沿着指定的维度进行拼接
503
0
0
解决AssertionError: size of input tensor and input format are different.tensor shape: (3, 138input_for
解决AssertionError: size of input tensor and input format are different.tensor shape: (3, 138input_for
480
0
0
|
Perl
成功解决ValueError: Cannot feed value of shape (1, 10, 4) for Tensor Placeholder:0 , which has shape
成功解决ValueError: Cannot feed value of shape (1, 10, 4) for Tensor Placeholder:0 , which has shape
526
0
0
热门文章
最新文章
1
评测：AI客服接入钉钉与微信的对比分析
2
Redis 基础
3
好程序员Python教程系列第5讲-分支结构
4
超酷Loading进度条
5
做强中国SaaS 八百客解读新老市场变化
6
Theano 中文文档 0.9 - 7.2.6 Theano如何处理形状信息
7
iOS - OC NSKeyedArchiver 数据归档
8
与众不同 windows phone (21) - Device（设备）之摄像头（拍摄照片, 录制视频）
9
REVERSE关键字之REVERSE函数
10
Http-Only is not secure [testing]
1
基于遗传优化的货柜货物摆放优化问题求解matlab仿真
42
2
基于遗传优化的Sugeno型模糊控制器设计matlab仿真
41
3
基于DVB-T的COFDM+16QAM+LDPC图传通信系统matlab仿真,包括载波同步,定时同步,信道估计
103
4
我的2024总结 | 降薪优化、焦虑生病、读书写作、逆势成长
49
5
Agent Laboratory：AI自动撰写论文，AMD开源自动完成科研全流程的多智能体框架
78
6
SVFR：全能视频人脸修复框架，支持提升清晰度、色彩填充和缺失补全等图像修复任务
48
7
PPTAgent：中科院开源AI工具，自动将文档转化为高质量PPT
104
8
深入理解 Flink 中的 State
44
9
StarTowerChain：去中心化匿名云存储，重塑数据存储未来
43
10
超强数据库管理软件推荐-没有之一-还在用Navicat管理本地数据库的吗？还在为Navicat寻求绿色版或者购买正版的费用望而却步吗？DBeaver让你解决所有数据库本地编写问题-优雅草央千澈-DBeaver下载和安装
40