Ajusted the code format and added an interface to swapaxes in the TensorVariable object.