[MXNet][原創]MXNet的加減乘除

A = mx.sym.Variable('A')
B = mx.sym.Variable('B')
C = A + B
D = A - B
E = A * B
F = A / B
# a = mx.nd.empty(1)  # 生成一個維度爲1的隨機值
a = mx.nd.array(2)
b = mx.nd.array(3)  # b等於3

executor = D.bind(ctx=mx.cpu(), args={'A': a, 'B': b})
print(type(executor))
# print(executor.arg_dict) 
executor.forward() 
print(executor.outputs[0])

 

輸出:

<class 'mxnet.executor.Executor'>

0.6666667
<NDArray  @cpu(0)>

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章