У меня есть тензор Кераса tens формы (Нет, Нет, 10) (включая размер партии).Я хотел бы разделить это на список тензоров формы (Нет, 10) на оси 1. Если бы я знал, что размер оси 1 был k, я мог бы сделать
layers.Lambda(lambda x: [x[:, i, :] for i in range(x.shape[1])])(tens)
Однако,потому что размер неизвестен, я получаю ошибку __index__ returned non-int (type NoneType) при построении этой модели.Как мне решить эту проблему?