def setup_benchmark(*args, **kwargs): a, b, axes = tensordot_npc.setup_benchmark(*args, **kwargs) axes_a, axes_b = axes non_axes_a = [i for i in range(a.rank) if i not in axes_a] non_axes_b = [i for i in range(b.rank) if i not in axes_b] return a.combine_legs((non_axes_a, axes_a)), b.combine_legs( (non_axes_b, axes_b))
def setup_benchmark(**kwargs): a, b, axes = tensordot_npc.setup_benchmark(**kwargs) axes_a, axes_b = axes axes_a = a.get_leg_indices(axes_a) axes_b = b.get_leg_indices(axes_b) return a.to_ndarray(), b.to_ndarray(), (axes_a, axes_b)
def setup_benchmark(**kwargs): a, b, axes = tensordot_npc.setup_benchmark(**kwargs) axes_a, axes_b = axes non_axes_a = [i for i in range(a.rank) if i not in axes_a] non_axes_b = [i for i in range(b.rank) if i not in axes_b] return a.to_ndarray(), b.to_ndarray(), ((non_axes_a, axes_a), (axes_b, non_axes_b))
def setup_benchmark(**kwargs): a, b, axes = tensordot_npc.setup_benchmark(**kwargs) axes_a, axes_b = axes non_axes_a = [i for i in range(a.rank) if i not in axes_a] non_axes_b = [i for i in range(b.rank) if i not in axes_b] a, pipes_a = combine_legs(a.to_ndarray(), (non_axes_a, axes_a)) b, pipes_b = combine_legs(b.to_ndarray(), (non_axes_b, axes_b)) return a, b, pipes_a, pipes_b