where \(\texttt{Mask}[k]\) is a vector of length \(d_{inter}\) with the \(k\)-th element as \(1\) and others as \(0\). For calculating \(\text{Imp}(W_{up}[:,k]|x)\) for all neurons in \(W_{up}\) parallelly, we introduce a diagonal mask matrix of size \((d_{inter},d_{inter})\), denoted as Mask.