jaxopt.tree_util.tree_vdot

jaxopt.tree_util.tree_vdot(tree_x, tree_y)[source]

Compute the inner product <tree_x, tree_y>.