pyvene.models.modeling_utils.bs_hd_to_bhsd

Contents

pyvene.models.modeling_utils.bs_hd_to_bhsd#

bs_hd_to_bhsd(tensor, h)[source]#

Convert a tensor of shape (b, s, h*d) back to (b, h, s, d).