Размазывание некоторой части тензора с помощью вектора маркировки 0-1
(я делаю некоторую работу по синтаксическому анализу c на основе BERT, и я собираюсь объединить несколько символов в слово вычислите их среднее значение или сумму и поместите один результат обратно в несколько позиций, где находятся символы ... 1007 * и вектор маркировки a: [0, 1, 1, 0, 1, 0], индексирующий слова для слияния, и ожидаемый результат должен заключаться в смазывании вложения 2-го и 3-го слова по оси 0
Кто-нибудь может помочь?
другое объяснение