2021年11月22日
pytorch的广播机制
摘要： 广播机制，就是将不同维度、不同长度的Tensor，在满足一定规则的前提下能够自动进行长度和维度的扩充，从而使不同维度、不同长度的Tensor之间正确的进行运算。 自动广播规则：两个tensor能够进行自动广播需要满足以下几个规则： 每个维度的大小都相等 每个Tensor至少有一个维度(不为空，如 ( 阅读全文
posted @ 2021-11-22 13:59 Liang-ml 阅读(405) 评论(0) 推荐(0)