ns</span>
<span class="sd">        -------</span>
<span class="sd">        list(np.ndarray), list(np.ndarray), list(np.ndarray)</span>

<span class="sd">        &quot;&quot;&quot;</span>
        <span class="n">block_size</span> <span class="o">=</span> <span class="n">unitary_mats</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>
        <span class="n">