Возвращает диагональную часть тензора.
Эта операция возвращает тензор с «диагональной» частью «входных данных». «Диагональная» часть вычисляется следующим образом:
Предположим, что «вход» имеет размерности «[D1,..., Dk, D1,..., Dk]», тогда выходной сигнал представляет собой тензор ранга «k» с размерностями «[D1,..., Dk]». где:
`диагональ[i1,..., ik] = input[i1, ..., ik, i1,..., ik]`.
Например:
# 'input' is [[1, 0, 0, 0]
[0, 2, 0, 0]
[0, 0, 3, 0]
[0, 0, 0, 4]]
tf.diag_part(input) ==> [1, 2, 3, 4]
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
static <T расширяет TType > TensorDiagPart <T> | |
Выход <Т> | диагональ () Выделенная диагональ. |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static TensorDiagPart <T> create (область области действия , ввод операнда <T>)
Фабричный метод для создания класса, обертывающего новую операцию TensorDiagPart.
Параметры
объем | текущий объем |
---|---|
вход | Тензор ранга k, где k четно, а не ноль. |
Возврат
- новый экземпляр TensorDiagPart