cascade.nn.multi_rbf

cascade.nn.multi_rbf(x, y)[source]

RBF kernel with support for multiplex dims

Parameters:

  • x (Tensor) – Input x of shape (*m, bs, n_vars)

  • y (Tensor) – Input y of shape (*m, bs, n_vars)

Return type:

Tensor

Returns:

RBF kernel of shape (*m, bs, bs)