Searched refs:flatten_to_dim (Results 1 – 1 of 1) sorted by relevance
| /dports/math/py-ssm/ssm-0.0.1/ssm/ |
| H A D | stats.py | 9 def flatten_to_dim(X, d): function
68 flat_L = flatten_to_dim(L, 2)
173 flat_data = flatten_to_dim(full_data, 1)
174 flat_mask = flatten_to_dim(full_mask, 1)
196 this_mus = flatten_to_dim(np.broadcast_to(this_mus, shp + (this_D,)), 1)[this_inds]
197 this_Ls = flatten_to_dim(np.broadcast_to(this_Ls, shp + (this_D, this_D)), 2)[this_inds]
|