(a, axes)
return a.permute(*axes)
Normal torch transpose is only valid for 2D matrices.
Normal torch transpose is only valid for 2D matrices.
Normal torch transpose is only valid for 2D matrices .
def transpose(a, axes):
    """Normal torch transpose is only valid for 2D matrices."""
    return a.permute(*axes)
def transpose ( a , axes ) : return a . permute ( * axes )