ninetoothed.debugging.simulate_arrangement

ninetoothed.debugging.simulate_arrangement(arrangement, tensors, device=None)

Simulate the arrangement of the tensors.

Parameters:

• arrangement – The arrangement of the tensors.

• tensors – The tensors.

• device – The device on which the tensors are.

Returns:

A tuple of source tensors and a tuple of target tensors, where each tuple is sorted according to the parameter order specified in arrangement, and each element in each tensor stores the index of that element in the source tensor.