diff --git a/src/controllers/dcg_noshare_controller.py b/src/controllers/dcg_noshare_controller.py index b27c2944976e39bdd16d5138081db91b43d3f42b..33afba71f05d5f2691f5b16f1586642d632fa167 100755 --- a/src/controllers/dcg_noshare_controller.py +++ b/src/controllers/dcg_noshare_controller.py @@ -68,7 +68,7 @@ class DCGnoshareMAC (DeepCoordinationGraphMAC): # Without decomposition, the payoff_fun output must only be reshaped output = output.view(*output.shape[:-1], n, n) # The output of the backward messages must be transposed - output[1] = output[1].transpose(dim0=-2, dim1=-1) + output[1] = output[1].clone().transpose(dim0=-2, dim1=-1) # Compute the symmetric average of each edge with it's flipped counterpart return output.mean(dim=0)