Pytorch / Numpy / python Эффективная интерполяция поврежденных данных - PullRequest
1 голос
/ 13 февраля 2020

У меня есть временная дискретная информация, которая может иметь пропущенные значения. У меня есть маска, указывающая, где находятся эти значения.

Как мне выполнить эффективную интерполяцию, заполняющую эти значения?

На практике у меня есть тензор TxCxJ ( Q ). Некоторые элементы, скажем, повреждены. Я хотел бы, учитывая поврежденный элемент Q[t,c,j], заполнить это значение интерполяцией между Q[t-1,c,j] и Q[t+1,c,j]

Кроме того, в худшем случае я могу найти несколько последовательных поврежденных элементов: Q[t_0:t_1,c,j]

заполняется интерполяцией между Q[t_0-1,c,j] и Q[t_1+1,c,j]

Можно использовать линейную интерполяцию, используя numpy или pytorch (или любую другую подходящую библиотеку, для которой я не Нужно учиться 5 месяцев: /). Я могу закодировать его с помощью циклов, но я искал какую-нибудь эффективную библиотеку, которая позволяла бы передавать маску или что-то вроде какой-нибудь классной индексации / маскировки, не требуя запуска алгоритма для известных точек.

Thaaanks

...