热线电话:13121318867

登录
2022-08-24 阅读量: 730
广播机制

当我们使用通用函数对两个数组进行计算时,通用函数会对这两个数组的对应元素进行计算,因此它要求这两个数组shape相同。如果shape不同的话,会进行如下的广播(broadcasting)处理

  • 让所有输入数组都向其中shape最长的数组看齐,shape中不足的部分都通过在前面加1补齐

  • 输出数组的shape是输入数组shape的各个轴上的最大值

  • 如果输入数组的某个轴和输出数组的对应轴的长度相同或者其长度为1时,这个数组能够用来计算,否则出错

  • 当输入数组的某个轴的长度为1时,沿着此轴运算时都用此轴上的第一组值


111.6075
0
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子