MxNet安装环境测试代码

import mxnet as mx

a = mx.nd.ones((2,3))

b = a * 2 + 1

b.asnumpy()

 

gpu版

import mxnet as mx

a = mx.nd.ones((2,3), mx.gpu())

b = a * 2 + 1

b.asnumpy()