I found that when passing lambda_path = [single_value], the shape of z is already missing the last dimension, therefore it already is 1D. I know passing a single value to lambda_path is not the "correct" thing to do, but for my specific case I need to work like this. For this I removed the squeeze part and directly return z as it is.
I found that when passing lambda_path = [single_value], the shape of z is already missing the last dimension, therefore it already is 1D. I know passing a single value to lambda_path is not the "correct" thing to do, but for my specific case I need to work like this. For this I removed the squeeze part and directly return z as it is.