来自 François Fleuret（@francoisfleuret）的推文线程

另一个（目前正在编写一个张量化的 RPN 内核） 给定一个 N × K 张量 X 和一个 N 张量 Y，对 X 的每一行中 Y 的前几个元素进行“旋转”，例如： X = [[1, 2, 3, 4], [10, 20, 30, 40]] Y = [ 2, 3 ] 结果 = [[2, 1, 3, 4], [20, 30, 10, 40]]