توی پایتون که هم به همان صورت بالا میشه نوشت و هم با تابع gather این تابع gather در c++ هم وجود داره توی c++ یک امکان دیگه هم وجود داره شما برای یک تنسور با متد flat_inner_dims به کلاس eigen::tensor دسترسی داشته باشید و از آنجا متد chp را استفاده کنید.