Я хочу индексировать последнюю ось тензора произвольной формы, за исключением последней, которая равна 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]
Мне не удалось найти какую-либо функцию тензорного потока или использование для нарезки произвольной формы.
Мне нужен общий метод индексации, чтобы я всегда мог получить доступ к последней оси для любого тензора формы.