e3nn_mlx.math.soft_one_hot_linspace

e3nn_mlx.math.soft_one_hot_linspace(x, start, end, number, basis, cutoff)[source]

Project values onto the normalized radial bases provided by upstream e3nn.

Parameters:
  • start (float)

  • end (float)

  • number (int)

  • basis (str)

  • cutoff (bool)