`iFddlZddlmZejddddZdZdS)N)_utilz T A, N r, N czN outzout = A != 0 ? r - c : 0cupyx_scipy_linalg_band_posc6tj|}|jdkrdStj||jdr|j}n|}|j\}}tjd|d|f\}}t|||}|jdrDttj |}ttj | }nCttj |}ttj | }||fS)aReturn the lower and upper bandwidth of a 2D numeric array. Parameters ---------- a : ndarray Input array of size (M, N) Returns ------- lu : tuple 2-tuple of ints indicating the lower and upper bandwidth. A zero denotes no sub- or super-diagonal on that side (triangular), and, say for M rows (M-1) means that side is full. Same example applies to the upper triangular part with (N-1). .. seealso:: :func:`scipy.linalg.bandwidth` r)rr F_CONTIGUOUS) cupyasarraysizer _assert_2dflagsTshapemgrid_kernel_cupy_band_pos_cintamaxamin) aAmnrow_numcol_numbandpts upper_band lower_bands s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/linalg/_array_utils.py bandwidthr s " QAv{{v Q w~ C  7DAqz!A#qs(+GW%a'::G w~.7++,, $)G,,--- 7++,, $)G,,--- z !!)r cupy.linalgrElementwiseKernelrrrrr"s\ 1$0 ! *"*"*"*"*"r