a dq@s*ddlZeejejdddZddZdS)N)ninputreturncCsZ|dkrtt||t|jdkr.t|jtj||j|jd}|d|jdddS)aReturn a 2-D tensor with ones on the diagonal and zeros elsewhere with the same batch size as the input. Args: n: the number of rows :math:`(N)`. input: image tensor that will determine the batch size of the output matrix. The expected shape is :math:`(B, *)`. Returns: The identity matrix with the same batch size as the input :math:`(B, N, N)`. rdevicedtypeN) AssertionErrortypelenshapetorchZeyerrrepeat)rridentityrZ/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/utils/misc.pyeye_likes  rcCs\|dkrtt||t|jdkr.t|jtj|d|j|jd}|d|jdddS)atReturn a 2-D tensor with a vector containing zeros with the same batch size as the input. Args: n: the number of rows :math:`(N)`. tensor: image tensor that will determine the batch size of the output matrix. The expected shape is :math:`(B, *)`. Returns: The vector with the same batch size as the input :math:`(B, N, 1)`. rrrN) r r r r r zerosrrr)rZtensorZvecrrrvec_likes  r)r intZTensorrrrrrrs