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