NDArray基本操作对NDArray的基本数学运算是元素粒度的:

# 创建两个全为1的NDArray
a = mx.nd.ones((2,3))
b = mx.nd.ones((2,3))

# 元素级加法
c = a + b

# 元素级减法
d = -c

# 元素级幂和正弦运算,然后转置
e = mx.nd.sin(c**2).T

# 元素级最大值
f = mx.nd.maximum(a, c)
f.asnumpy()