I. Introduction
The Terahertz (THz) bands, ranging from 0.1 THz to 10 THz, could provide an abundance of bandwidth and are promising for enabling wireless links with extreme capacity for six-generation (6G) communications and beyond [2], [3]. At THz frequencies, there exists noticeable signal attenuation, caused by spreading, molecular absorption, and diffusely scattering losses [4]. Fortunately, the sub-millimeter wavelength allows the incorporation of hundreds and even thousands of antenna elements into one antenna array, leading to ultra-massive multiple-input multiple-output (UM-MIMO) systems. These systems could assist in establishing highly directional links that efficiently combat the propagation losses in the THz band. Furthermore, with beam steering, THz UM-MIMO systems can reach Tbps data rates and enable massive connectivity [5].