05 2023 档案

SVD计算速度测试
摘要：```python import time import tensorflow as tf tf.compat.v1.disable_eager_execution() #%% 平均2.7秒。 发现conda 创建tf1.15 速度非常慢，应该环境配置有问题 A = tf.linalg.svd(tf 阅读全文
posted @ 2023-05-16 14:14 bregman 阅读(94) 评论(0) 推荐(0) 编辑
计算梯度jax
摘要：```python # %% import jax import jax.numpy as jnp import numpy as np def loss(params, r): lambda_a, lambda_s = params return jnp.maximum(r - lambda_a 阅读全文
posted @ 2023-05-13 13:26 bregman 阅读(35) 评论(0) 推荐(0) 编辑
L-BFGS-B算法
摘要：* 代码 https://github.com/nepluno/lbfgsb-gpu * https://pages.mtu.edu/~struther/Courses/OLD/5630/Refs/StdOpt/Par_L-BFGS-B_CompGraphFei_2014.pdf * jax 下 的 阅读全文
posted @ 2023-05-10 18:50 bregman 阅读(282) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示