那为什么B的形状不调整为(1,1,2)或者(2,1,1)?
通过广播机制,我们确保在操作的每个维度上,要么两个张量的维度相等,要么其中一个为 1。如果不满足这个条件,则无法进行广播。因此,为了使 𝐵B 能够与 𝐴A 进行逐元素运算,我们需要将 𝐵B 调整为 (1,2,1)(1,2,1),然后再进行广播。其他的调整方式(如 (1,1,2)(1,1,2) 或 (2,1,1)(2,1,1))在这个例子中不满足广播机制的规则,因而不可行。
那为什么B的形状不调整为(1,1,2)或者(2,1,1)?
通过广播机制,我们确保在操作的每个维度上,要么两个张量的维度相等,要么其中一个为 1。如果不满足这个条件,则无法进行广播。因此,为了使 𝐵B 能够与 𝐴A 进行逐元素运算,我们需要将 𝐵B 调整为 (1,2,1)(1,2,1),然后再进行广播。其他的调整方式(如 (1,1,2)(1,1,2) 或 (2,1,1)(2,1,1))在这个例子中不满足广播机制的规则,因而不可行。