CircularCorrelationCompositionModule

class CircularCorrelationCompositionModule(*args, **kwargs)[源代码]

基础:FunctionalCompositionModule

通过pykeen.nn.functional.circular_correlation()进行循环相关组合。

初始化内部模块状态,由nn.Module和ScriptModule共享。

方法总结

func(b)

计算两个向量之间的循环相关性。

方法文档

func(b: Tensor) Tensor

计算两个向量之间的循环相关性。

注意

实现使用了FFT。

Parameters:
  • a (Tensor) – 形状: s_1 包含第一个向量的张量。

  • b (Tensor) – 包含第二个向量的张量。

Returns:

向量之间的循环相关性。

Return type:

Tensor