cascade.nn.SpecNorm.dot

static SpecNorm.dot(x, y)[source]

Vector dot product with particles in the last dimension

Parameters:

  • x (Tensor) – Vector of shape (n_vars, n_particles)

  • y (Tensor) – Vector of shape (n_vars, n_particles)

Return type:

Tensor

Returns:

Dot product of shape (n_particles,)