美文网首页
mxnet:如何对多维NDArray按维度操作?

mxnet:如何对多维NDArray按维度操作?

作者: AI秘籍 | 来源:发表于2020-04-22 21:47 被阅读0次
    # 如何对多维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章节.

相关文章

网友评论

      本文标题:mxnet:如何对多维NDArray按维度操作?

      本文链接:https://www.haomeiwen.com/subject/beegihtx.html