我正在寻找一种在Tensorflow中实现块对角矩阵的方法。具体地说,我有一个块对角矩阵A,它有N个块,每个块的大小是S x S。此外,我有一个长度为N*S的向量v,我想计算A点v,在Tensorflow中有什么有效的方法吗?此外,我更喜欢支持批处理维度v的实现(例如,它的实维度是batch_size x (N*S)),并且它是内存高效的
我使用np.cov从二维数组中计算协方差矩阵,并利用它得到与Mahalanobis距离最近的邻居。site-packages/sklearn/externals/joblib/parallel.py:131: RuntimeWarning: invalid value encountered in sqrt如果我传递一个标识矩阵,而不是将c传递给NearestNeighbors,NearestNeighbors将按预期工作。我怀疑c实际上可能不是半正定的</