我想索引到具有任意形状的张量的最后一个轴，除了最后一个轴是 2。 例如，设 x 的形状为 （1,2,2）。索引到最后一个轴 x_0 = x[:, :, 0] # x_0, x_1 shapes a...