
    `i                         ddZ ddZdS )Tc                     |r|                                  |k    r| S |                                                     |          S )N)size
contiguousview)tensorr   check_same_sizes      s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/autograd/_functions/utils.py
maybe_viewr	      sA     6;;==D00##D)))    c                    |r|                                  |k    r| S |                                 t          |          z
  }d t          t	          |                                  |d          |                    D             }t          |          D ]}|                     dd          } |D ]}|                     |d          } | S )Nc                 *    g | ]\  }\  }}||k    |S  r   ).0dimexpandedoriginals       r   
<listcomp>z"maybe_unexpand.<locals>.<listcomp>   s:       %C%(H x	 	  r
       F)keepdimT)r   r   len	enumerateziprangesum)r   old_sizer   num_unsqueezedexpanded_dims_r   s          r   maybe_unexpandr   
   s     6;;==H44ZZ\\CMM1N )2noo.99*
 *
  M >"" . .Au-- / /C..Mr
   N)T)r	   r   r   r
   r   <module>r      s7   * * * *     r
   