摘要: opt_einsum, 爱因斯坦求和 jax, 算svd时比numpy 包快很多 flax 之前的一个问题, relu = max(x, 0) 在0点次梯度(不可导)的导数取了什么? 代码中是0。 import jax from jax.experimental import optimizers, 阅读全文
posted @ 2021-10-13 11:08 bregman 阅读(162) 评论(0) 推荐(0)