Удалить тензорные элементы по маске - PullRequest
0 голосов
/ 23 апреля 2020

Допустим, у меня есть тензор формы a = [Batchsize , 80 , 2048].
, и у меня есть другой тензор формы b = [Batchsize], содержащий 0, 1. Я хочу удалить элемент i из тензора a, который удовлетворяет b[i] = 1, и сохранить остальные тензоры в том же порядке.

Пока я делал это для циклов, но я уверен, что есть еще один быстрый способ сделать это. Спасибо

...