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).