广播法则(broadcast)

  广播法则(broadcast)是科学运算中经常使用的一个技巧,它在快速执行向量化的同时不会占用额外的内存/显存。 Numpy的广播法则(3条)定义如下: 让所有输入数组都向其中shape最长的数组看齐,shape中不足的部分通过在前面加1补齐 两个数组要么在某一个维度的长度一致,要么其中一个为1,否则不能计算 当输入数组的某个维度的长度为1时,计算时沿此维度复制扩充成一样的形状 PyTorch
相关文章
相关标签/搜索