a +=ic@sBdZddlmmZddZddZddZd d Zd d Z dS) z-Utility methods related to kernelized layers.NcCs8t|j}|dvr td||dkr4t|dS|S)zFIf input tensor is a vector (i.e., has rank 1), converts it to matrix.)z9The input tensor should have rank 1 or 2. Received rank: rr)lenshape ValueErrortf expand_dims)uZu_rankr m/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/keras/utils/kernelized_utils.py _to_matrixs  r cCst|}t|}|j}|j}|d|dkrJtd|dd|ddtt|dd|ddg}tt|d|dddg}||fS)zFAligns x and y tensors to allow computations over pairs of their rows.rzIThe outermost dimensions of the input tensors should match. Received y = z vs x = .r)r rrrZtiler)xyZx_matrixZy_matrixZx_shapeZy_shapeZx_tileZy_tiler r r _align_matrices!srcCs t|}t|}tj||ddS)NT)Z transpose_b)r rmatmul)r vr r r inner_product2srcCs<t||\}}ttj||d}t| d||S)a7Computes exact Gaussian kernel value(s) for tensors x and y and stddev. The Gaussian kernel for vectors u, v is defined as follows: K(u, v) = exp(-||u-v||^2 / (2* stddev^2)) where the norm is the l2-norm. x, y can be either vectors or matrices. If they are vectors, they must have the same dimension. If they are matrices, they must have the same number of columns. In the latter case, the method returns (as a matrix) K(u, v) values for all pairs (u, v) where u is a row from x and v is a row from y. Args: x: a tensor of rank 1 or 2. It's shape should be either [dim] or [m, dim]. y: a tensor of rank 1 or 2. It's shape should be either [dim] or [n, dim]. stddev: The width of the Gaussian kernel. Returns: A single value (scalar) with shape (1, 1) (if x, y are vectors) or a matrix of shape (m, n) with entries K(u, v) (where K is the Gaussian kernel) for all (u,v) pairs where u, v are rows from x and y respectively. Raises: ValueError: if the shapes of x, y are not compatible. r)rr reduce_summathZsquared_differenceexp)rrstddev x_aligned y_alignedZdiff_squared_l2_normr r r exact_gaussian_kernel8s rcCs8t||\}}ttt||d}t| |S)a2Computes exact Laplacian kernel value(s) for tensors x and y using stddev. The Laplacian kernel for vectors u, v is defined as follows: K(u, v) = exp(-||u-v|| / stddev) where the norm is the l1-norm. x, y can be either vectors or matrices. If they are vectors, they must have the same dimension. If they are matrices, they must have the same number of columns. In the latter case, the method returns (as a matrix) K(u, v) values for all pairs (u, v) where u is a row from x and v is a row from y. Args: x: a tensor of rank 1 or 2. It's shape should be either [dim] or [m, dim]. y: a tensor of rank 1 or 2. It's shape should be either [dim] or [n, dim]. stddev: The width of the Gaussian kernel. Returns: A single value (scalar) with shape (1, 1) if x, y are vectors or a matrix of shape (m, n) with entries K(u, v) (where K is the Laplacian kernel) for all (u,v) pairs where u, v are rows from x and y respectively. Raises: ValueError: if the shapes of x, y are not compatible. r)rrrabssubtractr)rrrrrZ diff_l1_normr r r exact_laplacian_kernelWsr) __doc__Ztensorflow.compat.v2compatv2rr rrrrr r r r s