# 如何对多维NDArray按维度操作?
    X = nd.array([[1, 2, 3], [4, 5, 6]])
    # 对同⼀列（axis=0）的元素求和，并在结果中保留⾏和列这两个维度（keepdims=True）
    print(X.sum(axis=0, keepdims=True))
    # 对同⼀⾏（axis=1）的元素求和，并在结果中保留⾏和列这两个维度（keepdims=True）
    print(X.sum(axis=1, keepdims=True))

image.png
image.png

参考:

1. 动手学深度学习第3.6.3章节.