Скопируйте тензор, устанавливающий все за пределами центральной полосы в каждой самой внутренней матрице в ноль.
Часть `band` вычисляется следующим образом: предположим, что `вход` имеет `k` измерений `[I, J, K, ..., M, N]`, тогда выходной сигнал представляет собой тензор той же формы, где
`band[i, j, k, ..., m, n] = in_band(m, n) * input[i, j, k, ..., m, n]`.
Индикаторная функция
`in_band(m, n) = (num_lower < 0 || (mn) <= num_lower)) && (num_upper < 0 || (nm) <= num_upper)`.
Например:
# if 'input' is [[ 0, 1, 2, 3]
[-1, 0, 1, 2]
[-2, -1, 0, 1]
[-3, -2, -1, 0]],
tf.matrix_band_part(input, 1, -1) ==> [[ 0, 1, 2, 3]
[-1, 0, 1, 2]
[ 0, -1, 0, 1]
[ 0, 0, -1, 0]],
tf.matrix_band_part(input, 2, 1) ==> [[ 0, 1, 0, 0]
[-1, 0, 1, 0]
[-2, -1, 0, 1]
[ 0, -2, -1, 0]]
tf.matrix_band_part(input, 0, -1) ==> Upper triangular part.
tf.matrix_band_part(input, -1, 0) ==> Lower triangular part.
tf.matrix_band_part(input, 0, 0) ==> Diagonal.
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Унаследованные методы
логическое значение | равно (Объект arg0) |
последний класс<?> | получитьКласс () |
интервал | хэш-код () |
окончательная пустота | уведомить () |
окончательная пустота | уведомитьВсе () |
Нить | toString () |
окончательная пустота | подождать (длинный arg0, int arg1) |
окончательная пустота | подождите (длинный arg0) |
окончательная пустота | ждать () |
абстрактная среда выполнения | окружение () Верните среду выполнения, в которой была создана эта операция. |
абстрактная операция |
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
публичный выходной диапазон <T> ()
Тензор ранга `k` той же формы, что и входные данные. Извлеченный полосатый тензор.
public static BandPart <T> create (область действия , ввод операнда <T>, операнд <U> numLower, операнд <U> numUpper)
Фабричный метод для создания класса, обертывающего новую операцию BandPart.
Параметры
объем | текущий объем |
---|---|
вход | Тензор ранга `k`. |
numLower | 0-D тензор. Количество сохраняемых субдиагоналей. Если отрицательно, сохраните весь нижний треугольник. |
numUpper | 0-D тензор. Количество сохраняемых супердиагоналей. Если отрицательно, сохраните весь верхний треугольник. |
Возврат
- новый экземпляр BandPart