espaloma.mm.functional.periodic

espaloma.mm.functional.periodic(x, k, periodicity=[1, 2, 3, 4, 5, 6], phases=[0.0, 0.0, 0.0, 0.0, 0.0, 0.0])

Periodic term.

Parameters

• x (torch.Tensor, shape=(batch_size, 1))

• k (torch.Tensor, shape=(batch_size, number_of_phases))

• periodicity (either list of length number_of_phases, or) – torch.Tensor, shape=(batch_size, number_of_phases)

• phases (either list of length number_of_phases, or) – torch.Tensor, shape=(batch_size, number_of_phases)