Searched refs:axes_to_sum (Results 1 – 1 of 1) sorted by relevance
794 axes_to_sum = (800 axes_to_sum)901 def _einsum_v1_reduction(t0, t0_axis_labels, t1, t1_axis_labels, axes_to_sum): argument950 assert all(a in t0_axis_labels and a in t1_axis_labels for a in axes_to_sum)951 preserved_axes = (set(t0_axis_labels) & set(t1_axis_labels)) - axes_to_sum954 broadcast_axes[i] = set(sym_list) - preserved_axes - axes_to_sum964 (input_index == 1 and a in axes_to_sum)):980 if not axes_to_sum:999 t0_shape[len(preserved_axes):-len(axes_to_sum)])1000 num_summed_elements = _total_size(t0_shape[-len(axes_to_sum):])[all …]