阵列广播能否将较大的维度视为较小的数字?

我试图通过使用阵列广播来避免for循环,以节省时间。

我的设置如下:我有一个长度为的列数组 n 其中每个元素是 mxm 数组(我称之为 A )

和 nxm 数组(我称之为 E )

我的最终目标是

其中,这个新数组的每一行都是a_i(a)之间数组广播的结果 mxm 数组)及其在E中的对应行(具有t_i的行)。

看起来正常的阵列广播规则不会允许这种情况发生,因为从技术上讲 A. 是a nmxm 阵列,我不相信,可以用它来广播 E 自从 E 是 nxm 但你也可以考虑 A. 作为一个 nx1 列中的每个元素都是 mxm 数组,则维度将在数组广播规则范围内。有没有办法迫使这种情况发生?

此外,在我的实际代码中,每个 A_i 是一样的。也许这会让事情变得更容易。