MathTop

array_scan_along_dimension 函数

沿选定维度计算前缀归约，并保持原数组形状。

语法

  • ArrayTools:-ScanAlongDimension(reducer, array)
  • ArrayTools:-ScanAlongDimension(reducer, array, dimension)
  • array_scan_along_dimension(reducer, array, dimension)

参数与选项

reducer归约器名称：add、sum、multiply、product、min 或 max。
array要执行前缀扫描的矩形数组。
dimension可选一基维度；省略时使用第一个长度大于 1 的维度。

示例

  1. 列方向前缀和ArrayTools:-ScanAlongDimension(add, [[1, 2, 3], [4, 5, 6]], 1)返回 [[1, 2, 3], [5, 7, 9]]。
  2. 行方向前缀最大值ArrayTools:-ScanAlongDimension(max, [[1, 5, 3], [4, 2, 6]], 2)返回 [[1, 5, 5], [4, 4, 6]]。