python中的广播机制

当对两个形状不同的 Tensor 按元素运算 时,可能会触发⼴播(broadcasting)机制:先适当复制元素使这两个 Tensor 形状相同后再按元素运算

 

(x 中第⼀⾏的2个元素被⼴播 (复制)到了第⼆⾏和第三⾏,⽽ y 中第⼀列的3个元素被⼴播(复制)到了第⼆列)

 

posted @ 2020-08-21 21:34  Jason66661010  阅读(677)  评论(0编辑  收藏  举报