Is there any function, can get the last embedding ? eg: [245, 1, 256] -> only take the 256 embedding list , not all embedding 245x256 best regard