Применение ядер для разделения измерений в gpytorch

Question

Я перевожу трехмерный код регрессии гауссовского процесса из GPflow в GPyTorch, чтобы воспользоваться встроенной масштабируемостью PyTorch для точного георадара.

Этот пример полезен, но не может объяснение того, как применять отдельные ядра к отдельным измерениям в многомерном георадаре (например, ядро ​​Matern52 для измерений [0,1] и ядро ​​Matern 32 для измерения [2]).

Есть ли GPyTorch поддерживают эту функциональность, и если да, то как она реализована / где найти пример?

Answer 1

Используйте active_dims в функциях ядра:

from (https://docs.gpytorch.ai/en/v1.1.1/kernels.html#rbfkernel) active_dims (кортеж целых чисел, необязательно): установите это, если вы хотите вычислить ковариацию только несколько входных размеров. Цифры соответствуют индексам размеров. По умолчанию: Нет.