если у меня есть тензор, форма [none, none] в тензорном потоке (нет palceholder, например batch_size), тогда я хочу получить максимальный элемент, кроме элементов diag, как я могу его получить