Helper for calculating broadcast shapes with core dimensions.
(broadcast_shape, dim_sizes, list_of_core_dims)
| 2248 | |
| 2249 | |
| 2250 | def _calculate_shapes(broadcast_shape, dim_sizes, list_of_core_dims): |
| 2251 | """Helper for calculating broadcast shapes with core dimensions.""" |
| 2252 | return [broadcast_shape + tuple(dim_sizes[dim] for dim in core_dims) |
| 2253 | for core_dims in list_of_core_dims] |
| 2254 | |
| 2255 | |
| 2256 | def _create_arrays(broadcast_shape, dim_sizes, list_of_core_dims, dtypes, |
no outgoing calls
no test coverage detected
searching dependent graphs…