在gpytorch中应用内核来分离维度

Question

我正在将三维高斯过程回归代码从GPflow转换为GPyTorch，以利用PyTorch的本地可伸缩性进行精确的探地雷达。

This example是有帮助的，但无法解释如何将不同的内核应用于多维探地雷达中的各个维度(例如，将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 (整数元组，可选)：如果只想计算几个输入维度的协方差，则设置此选项。整数对应于维度的索引。默认值:无。