建议开发者统一接口参数调用

比如 concatenate:
https://mxnet.incubator.apache.org/api/python/ndarray/ndarray.html?highlight=flatten#mxnet.ndarray.concatenate

第二个参数是 axis

concat:
https://mxnet.incubator.apache.org/api/python/ndarray/ndarray.html?highlight=flatten#mxnet.ndarray.concatenate

维度参数变成了 dim

numpy 中相同操作的 api 维度参数是 axis: https://docs.scipy.org/doc/numpy/reference/generated/numpy.concatenate.html#numpy.concatenate

还有许多这种例子

隔壁 pytorch 自家的 tensor 都是 dim 取维度

2赞

有些函数是要用shape=()来指定维度,有些又可以直接用两个参数写维度,不太统一,要统一下最好