o i@s8ddlZdejdedejfddZddZd d d ZdS) Nxn_rowsreturncCsF|jd|d}|}tj||jdd||f||dfdS)z%return a view of x as a Hankel matrixN)sizestride)shapertorch as_strided)rrn_cols x_stridesrN/home/ubuntu/.local/lib/python3.10/site-packages/fast_bss_eval/torch/linalg.py hankel_views "rcCsHtj|dddfjdd|fdd}t||jdd}|jd d}|S) z( make symmetric toeplitz matrix .rN)rdimsrdim)r)r catfliprr )rtwo_xHTrrrtoeplitz!s& rc Cs|jdd\}}}|dur|d}||jdd||||f}t|jdD]>}t|jdD]4}tj|d| dd||f|dd|||ffdd}t||jd d } | |ddd|dd|f<q2q)||jdd ||||f}|S) zJ Create a block matrix where the blocks have a Toeplitz structure Nrr.rrrr)r new_zerosranger rrrreshape) rntoprowscolsoutrcvecrrrrblock_toeplitz+s 8$r+)N)r Tensorintrrr+rrrrs