Как векторизовать это в tenorflow?

Давайте день A - это матрица M x J, а B - матрица N x J.

Я хотел бы создать матрицу M x N S, где:

S_ij = w^t[A_i;B_j;A_i o B_j)  

По сути, каждый элемент матрицы результатов S представляет собой некоторый вектор w (строка матрицы W) с точечным произведением: col A_j, совпадающий с col B_j и конкрементный с поэлементным произведением каждого A_i и B_j для всех i и j.

В идеале я хотел бы векторизовать операцию и использовать только матричные формы S, A и B вместо разрезания и выполнения для циклов. Однако я новичок в tenorflow и, похоже, не могу понять, как сделать код в матричной форме, поскольку A_i o B_j, кажется, приводит к 2d матрице, и он должен быть согласован с вектором A_i и B_j.

Спасибо

0 ответов

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