[Bart] Question for BartModel Output shape

It worked. Thank you very much!