Nn Ops

Выполняет усредняющее пулингирование на входе.

Выполняет усредняющее пулингирование 3D на входе.

Вычисляет градиенты функции усредняющего пулингирования.

Добавляет bias к value.

Обратная операция для "BiasAdd" по тензору "bias".

Вычисляет 2D свёртку, учитывая 4-мерные input и filter тензоры.

Вычисляет градиенты свёртки по фильтру.

Вычисляет градиенты свёртки по входу.

Вычисляет 3D свёртку, учитывая 5-мерные input и filter тензоры.

Вычисляет градиенты 3D свёртки по фильтру.

Вычисляет градиенты 3D свёртки по входу.

Возвращает индекс измерения в целевом формате данных, учитывая индекс в исходном.

Возвращает пермутированный вектор/тензор в целевом формате данных, учитывая исходный.

Вычисляет 2D свёртку по глубине, учитывая 4-мерные input и filter тензоры.

Вычисляет градиенты свёртки по глубине по фильтру.

Вычисляет градиенты свёртки по глубине по входу.

Вычисляет градации серого 4-мерных input и 3-мерных filter тензоров.

Вычисляет градиент морфологической 2D дилатации по фильтру.

Вычисляет градиент морфологической 2D дилатации по входу.

Вычисляет экспоненциальную линейную функцию: exp(features) - 1 если < 0, features в противном случае.

Выполняет дробное усредняющее пулингирование на входе.

Выполняет дробное максимальное пулингирование на входе.

Нормализация по батчам.

Градиент для нормализации по батчам.

Градиент для нормализации по батчам.

Градиент для нормализации по батчам.

Нормализация по батчам.

Нормализация по батчам.

Выполняет заполнение как предподготовку во время свёртки.

Выполняет изменение размера и заполнение как предподготовку во время свёртки.

Указывает, находятся ли целевые значения в ТОП K прогнозов.

Указывает, находятся ли целевые значения в ТОП K прогнозов.

L2 потеря.

Локальная нормализация отклика.

Вычисляет логистическую софтмакс-активацию.

Выполняет максимальное пулингирование на входе.

Выполняет максимальное пулингирование 3D на входе.

Вычисляет градиенты функции максимального пулингирования 3D.

Вычисляет градиенты второго порядка функции максимального пулингирования.

Вычисляет градиенты второго порядка функции максимального пулингирования.

Вычисляет градиенты второго порядка функции максимального пулингирования.

Вычисляет градиенты второго порядка функции максимального пулингирования.

Вычисляет градиенты функции максимального пулингирования.

Выполняет максимальное пулингирование на входе.

Выполняет максимальное пулингирование на входе и выводит как максимальные значения, так и их индексы.

Находит значения n-го порядка статистики для последнего измерения.

Производит среднее пулингирование входного тензора для квантованных типов.

Квантованная нормализация по батчам.

Добавляет тензор 'bias' к тензору 'input' для квантованных типов.

Вычисляет 2D свёртку, учитывая квантованные 4D входные и фильтры.

Производит максимальное пулингирование входного тензора для квантованных типов.

Вычисляет квантованное выпрямляющее линейное преобразование: max(features, 0)

Вычисляет квантованное выпрямляющее линейное преобразование 6: min(max(features, 0), 6)

Вычисляет квантованное выпрямляющее линейное преобразование X: min(max(features, 0), max_value)

Вычисляет выпрямляющее линейное преобразование: max(features, 0).

Вычисляет выпрямляющее линейное преобразование 6: min(max(features, 0), 6).

Вычисляет масштабированное экспоненциальное линейное преобразование: scale * alpha * (exp(features) - 1)

Вычисляет софтмакс-активацию.

Вычисляет стоимость перекрестной энтропии софтмакса и градиенты для обратного распространения.

Вычисляет софтплюс: log(exp(features) + 1).

Вычисляет софтсигн: features / (abs(features) + 1).

Вычисляет стоимость перекрестной энтропии софтмакса и градиенты для обратного распространения.

Находит значения и индексы k наибольших элементов для последнего измерения.