espaloma.mm.functional.harmonic

espaloma.mm.functional.harmonic(x, k, eq, order=[2])

Harmonic term.

Parameters:

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

• k (torch.Tensor, shape=(batch_size, len(order)))

• eq (torch.Tensor, shape=(batch_size, len(order)))

• order (int or List of int)

Returns:

u

Return type:

torch.Tensor, shape=(batch_size, 1)