Spec-Zone.ru › TensorFlow C++

tensorflow::ops::Conv

#include <nn_ops.h>

Вычисляет N-мерную свёртку для тензоров input размерности (N+1+batch_dims) и filter размерности (N+2).

Краткое описание

Общая функция для вычисления N-мерной свёртки. Необходимо, чтобы выполнялось 1 <= N <= 3.

Аргументы:

  • scope: объект Scope
  • input: Tensor типа T и формы batch_shape + spatial_shape + [in_channels] в случае, если channels_last_format = true, или формы batch_shape + [in_channels] + spatial_shape, если channels_last_format = false. spatial_shape имеет N измерений с N=2 или N=3. Также обратите внимание, что batch_shape определяется параметром batch_dims и по умолчанию равно 1.
  • filter: (N+2)-DTensor того же типа, что и input, и формы spatial_filter_shape + [in_channels, out_channels], где spatial_filter_shape имеет N измерений с N=2 или N=3.
  • strides: одномерный тензор длины N+2. Шаг скользящего окна для каждого измерения input. Должен содержать strides[0] = strides[N+1] = 1.
  • padding: Тип используемого алгоритма дополнения.

Необязательные атрибуты (см. Attrs):

  • explicit_paddings: Если padding равно "EXPLICIT", задаёт список явных величин дополнения. Для i-го измерения величина дополнения, добавляемого до и после измерения, равна соответственно explicit_paddings[2 * i] и explicit_paddings[2 * i + 1]. Если padding не равно "EXPLICIT", значение explicit_paddings должно быть пустым.
  • data_format: Используется для задания формата данных. По умолчанию CHANNELS_FIRST используется NHWC (2D) / NDHWC (3D) или, если CHANNELS_LAST, используется NCHW (2D) / NCDHW (3D).
  • dilations: одномерный тензор длины N+2. Коэффициент разрежения для каждого измерения input. Если задано значение k > 1, между элементами фильтра в этом измерении будут пропущены k-1 ячеек. Порядок измерений определяется значением channels_last_format; подробности см. выше. Коэффициенты разрежения для измерений пакета и глубины должны быть равны 1.
  • batch_dims: Положительное целое число, задающее количество измерений пакета входного тензора. Оно должно быть меньше ранга входного тензора.
  • groups: Положительное целое число, задающее количество групп, на которые разбивается входной тензор вдоль оси каналов. Каждая группа свёртывается отдельно с фильтрами filters / groups. Результаты всех групп объединяются вдоль оси каналов. Количество каналов Input и фильтров должно делиться на groups без остатка.

Возвращает:

  • Output: тензор размерности (N+1+batch_dims). Порядок измерений определяется значением channels_last_format; подробности см. ниже.
Конструкторы и деструкторы
Conv(const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input filter, const gtl::ArraySlice< int > & strides, StringPiece padding)
Conv(const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input filter, const gtl::ArraySlice< int > & strides, StringPiece padding, const Conv::Attrs & attrs)
Открытые атрибуты
operation
Operation
output
::tensorflow::Output
Открытые функции
node() const
::tensorflow::Node *
operator::tensorflow::Input() const
operator::tensorflow::Output() const
Открытые статические функции
BatchDims(int64 x)
Attrs
DataFormat(StringPiece x)
Attrs
Dilations(const gtl::ArraySlice< int > & x)
Attrs
ExplicitPaddings(const gtl::ArraySlice< int > & x)
Attrs
Groups(int64 x)
Attrs
Структуры
tensorflow::ops::Conv::Attrs

Методы задания необязательных атрибутов для Conv.

Открытые атрибуты

operation

Operation operation

output

::tensorflow::Output output

Открытые функции

Conv

 Conv(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input filter,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

Conv

 Conv(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input filter,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding,
  const Conv::Attrs & attrs
)

node

::tensorflow::Node * node() const 

operator::tensorflow::Input

operator::tensorflow::Input() const 

operator::tensorflow::Output

operator::tensorflow::Output() const 

Открытые статические функции

BatchDims

Attrs BatchDims(
  int64 x
)

DataFormat

Attrs DataFormat(
  StringPiece x
)

Dilations

Attrs Dilations(
  const gtl::ArraySlice< int > & x
)

ExplicitPaddings

Attrs ExplicitPaddings(
  const gtl::ArraySlice< int > & x
)

Groups

Attrs Groups(
  int64 x
)

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/conv

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API