pyvene.models.modeling_utils.bhsd_to_bs_hd# bhsd_to_bs_hd(tensor)[source]# Convert a tensor of shape (b, h, s, d) to (b, s, h*d).