问题描述

张量A和B在广播过程中，对应维度不匹配。

张量广播

PyTorch操作支持广播，其张量参数可以自动扩展为相同大小（不复制数据）。
PyTorch广播语义密切跟随numpy式广播。如果你熟悉数字广播，可以按照之前流程执行。
其广播机制遵循以下规则：

  1. 每个张量至少具有一个维度；
  2. 广播时，从尾部维度开始，满足尺寸大小相等或其中一个为1或其中一个不存在。

正例：
在这里插入图片描述

反例：
在这里插入图片描述

Reference

http://www.dataguru.cn/article-11907-1.html

Logo
GitCode 开源社区

旨在为数千万中国开发者提供一个无缝且高效的云端环境，以支持学习、使用和贡献开源项目。

更多推荐

cover

GitTalk | 使用面向业务的狮偶编程语言提升开发效率

avatar GitCode 开源社区
cover

GitTalk | DevUI Suits 场景解决方案

avatar GitCode 开源社区
cover

GitTalk | DevUI Admin 前端项目构建

avatar GitCode 开源社区