Amortized Eigendecomposition for Neural Networks