我想索引到任意形状张量的最后一个轴，除了最后一个轴是2。
。设x的形状为(1,2,2)。
到最后一个轴的索引x_0 = x[:, :, 0] # x_0, x_1 shapes are (1,2)
x_1 = x[:, :, 1]
。设x的形状为(1,2,3,4,2)。
到最后一个轴的索引x_0 = x[:, :, :, :, 0] # x_0, x_1 shapes are (1,2,3,4)
x_1 = x[:, :, :, :, 1]
我一直无法找到任何张sorflow函数或用于切割任意形状的用法。
我需要一个通用的索引方法，这样我就可以访问任何形状张量的最后一个轴。
tensorflow中的切片语法与numpy非常相似。在这种情况下可以使用省略号:
Ellipsis扩展为选择元组索引所有维度所需的
:对象的数量。在大多数情况下，这意味着扩展后的选择元组的长度为x.ndim。只能有一个省略号
对你来说，
x_0 = x[..., 0]
将索引任意形状张量的最后一个轴。
您还可以查看以下问题的答案:numpy中的slice(:)和ellipsis(…)操作符之间的区别是什么?