pyvene.models.constants.split_head_and_permute

pyvene.models.constants.split_head_and_permute#

split_head_and_permute(x, num_head)#