Bering.models.GaussianSmearing.forward

GaussianSmearing.forward(x, y)

Compute smeared gaussian features between data.

Parameters:

x (Tensor): data of shape \((..., d)\) y (Tensor): data of shape \((..., d)\)

Returns:

Tensor: features of shape \((..., num\_kernel)\)