тензорный поток:: опс:: Средний пул3D
#include <nn_ops.h>
Выполняет 3D-усреднение на входе.
Краткое содержание
Аргументы:
- область: объект области.
- ввод: Форма тензора
[batch, depth, rows, cols, channels]
для объединения. - ksize: 1D тензор длины 5. Размер окна для каждого измерения входного тензора. Должно быть
ksize[0] = ksize[4] = 1
. - шаги: 1-D тензор длины 5. Шаг скользящего окна для каждого измерения
input
. Должно бытьstrides[0] = strides[4] = 1
. - дополнение: тип используемого алгоритма заполнения.
Необязательные атрибуты (см. Attrs
):
- data_format: формат входных и выходных данных. В формате по умолчанию «NDHWC» данные сохраняются в следующем порядке: [пакет, глубина_внутри, высота_внутри, ширина_в_каналах]. Альтернативно формат может быть «NCDHW», порядок хранения данных: [пакет, in_channels, in_глубина, in_высота, in_ширина].
Возврат:
-
Output
: средний объединенный выходной тензор.
Конструкторы и деструкторы | |
---|---|
AvgPool3D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding) | |
AvgPool3D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const AvgPool3D::Attrs & attrs) |
Публичные атрибуты | |
---|---|
operation | |
output |
Общественные функции | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Публичные статические функции | |
---|---|
DataFormat (StringPiece x) |
Структуры | |
---|---|
tensorflow:: ops:: AvgPool3D:: Attrs | Дополнительные установщики атрибутов для AvgPool3D . |
Публичные атрибуты
операция
Operation operation
выход
::tensorflow::Output output
Общественные функции
Средний пул3D
AvgPool3D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding )
Средний пул3D
AvgPool3D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const AvgPool3D::Attrs & attrs )
узел
::tensorflow::Node * node() const
оператор::tensorflow::Input
operator::tensorflow::Input() const
оператор::tensorflow::Выход
operator::tensorflow::Output() const
Публичные статические функции
Формат данных
Attrs DataFormat( StringPiece x )