Ковариационная матрица из 2D векторов - Tensorflow, Numpy

Я пытаюсь сгенерировать функцию ядра для GP, используя только операции Matrix (без циклов). Векторы, в которых без проблем можно воспользоваться вещанием

def kernel(A,B):
    return 1/np.exp(np.linalg.norm(A-B.T))**2

A и B оба являются [n,1] векторами, но с матрицами в форме [n,m] это просто не работает. (Также попытался изменить форму на [1,n,m]). Мне интересно вычислить матрицу X, в которой каждый ij-й элемент определяется Ai-Bj.

Сейчас я работаю над Numpy, но моя конечная цель - реализовать это на Tensorflow. Заранее спасибо.

0 ответов

Другие вопросы по тегам