pymiediff.special.pi_tau

pymiediff.special.pi_tau(n: int, mu: Tensor, **kwargs)

Compute angular functions pi_n and tau_n up to order n.

Parameters:

• n (int or torch.Tensor) – Maximum order.

• mu (torch.Tensor) – Cosine of polar angle.

Returns:

(pi, tau) with order axis first (including order 0).

Return type:

tuple of torch.Tensor