Skip to content
Snippets Groups Projects
Commit 2e5ec5b5 authored by Gordian Edenhofer's avatar Gordian Edenhofer
Browse files

lanczos_tridiag: Add potential TODOs as comment

parent 0a3987b6
Branches
Tags
1 merge request!832Better Lanczos interface
...@@ -26,6 +26,10 @@ def lanczos_tridiag( ...@@ -26,6 +26,10 @@ def lanczos_tridiag(
v = v / jnp.linalg.norm(v) v = v / jnp.linalg.norm(v)
vecs = vecs.at[0].set(v) vecs = vecs.at[0].set(v)
# TODO
# * use `forest_util.dot` in favor of plain `jnp.dot`
# * remove all reshapes as they are unnecessary
# Zeroth iteration # Zeroth iteration
w = mat(v) w = mat(v)
if w.shape != shape_dtype_struct.shape: if w.shape != shape_dtype_struct.shape:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment