jax代替numpy提升运算速度
经办公室小伙伴介绍,jax包代替numpy可大大提升运算速度例子## pip install jax## pip install jaxlibfrom jax import random, jitimport jax.numpy as jnpx=np.random.randn(2000000)@jit ###调用jnp前必有def abc(x): a=jnp.log10(x) b=10**a c=jnp.sum(b) return b
原创
2021-03-30 15:27:29 ·
708 阅读 ·
0 评论