Skip to content

Commit

Permalink
remove \sum from docstring --> generates error
Browse files Browse the repository at this point in the history
  • Loading branch information
twallema committed Jul 25, 2024
1 parent 283218f commit d413534
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tutorials/IDD/spatial_SIR/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@ def matmul_2D_3D_matrix(X, W):
======
X_out : np.ndarray
Matrix product of size (n, m).
Element-wise equivalent operation: O_{ij} = \sum_{l} [ s_{il} * w_{lji} ]
Element-wise equivalent operation: O_{ij} = sum_{l} [ s_{il} * w_{lji} ]
"""
W = np.atleast_3d(W)
return np.einsum('ik,kji->ij', X, np.broadcast_to(W, (W.shape[0], W.shape[0], X.shape[0])))

0 comments on commit d413534

Please sign in to comment.