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