Helper for calculating broadcast shapes with core dimensions.
(broadcast_shape, dim_sizes, list_of_core_dims)
| 2093 | |
| 2094 | |
| 2095 | def _calculate_shapes(broadcast_shape, dim_sizes, list_of_core_dims): |
| 2096 | """Helper for calculating broadcast shapes with core dimensions.""" |
| 2097 | return [broadcast_shape + tuple(dim_sizes[dim] for dim in core_dims) |
| 2098 | for core_dims in list_of_core_dims] |
| 2099 | |
| 2100 | |
| 2101 | def _create_arrays(broadcast_shape, dim_sizes, list_of_core_dims, dtypes, |
no outgoing calls
no test coverage detected