dòng chảy căng:: ôi:: MaxPoolWithArgmax

#include <nn_ops.h>

Thực hiện gộp tối đa trên đầu vào và đầu ra cả giá trị và chỉ số tối đa.

Bản tóm tắt

Các chỉ số trong argmax được làm phẳng, sao cho giá trị tối đa tại vị trí [b, y, x, c] trở thành chỉ mục được làm phẳng: (y * width + x) * channels + c nếu include_batch_in_index là Sai; ((b * height + y) * width + x) * channels + c nếu include_batch_in_index là Đúng.

Các chỉ số được trả về luôn ở dạng [0, height) x [0, width) trước khi làm phẳng, ngay cả khi có liên quan đến phần đệm và câu trả lời đúng về mặt toán học nằm ở bên ngoài (âm hoặc quá lớn). Đây là một lỗi nhưng việc khắc phục nó rất khó thực hiện theo cách tương thích ngược an toàn, đặc biệt là do bị làm phẳng.

Lập luận:

  • phạm vi: Một đối tượng Phạm vi
  • đầu vào: 4-D với hình dạng [batch, height, width, channels] . Đầu vào để gộp lại.
  • ksize: Kích thước của cửa sổ cho từng chiều của tensor đầu vào.
  • bước tiến: Bước tiến của cửa sổ trượt cho từng chiều của tensor đầu vào.
  • phần đệm: Loại thuật toán đệm sẽ sử dụng.

Thuộc tính tùy chọn (xem Attrs ):

  • include_batch_in_index: Có bao gồm thứ nguyên lô trong chỉ mục phẳng của argmax hay không.

Trả về:

  • Đầu ra Output ra: Tenor đầu ra gộp tối đa.
  • Output argmax: 4-D. Các chỉ số làm phẳng của các giá trị tối đa được chọn cho mỗi đầu ra.

Hàm tạo và hàm hủy

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

Thuộc tính công khai

argmax
operation
output

Các hàm tĩnh công khai

IncludeBatchInIndex (bool x)
Targmax (DataType x)

Cấu trúc

tensorflow:: ops:: MaxPoolWithArgmax:: Attrs

Trình thiết lập thuộc tính tùy chọn cho MaxPoolWithArgmax .

Thuộc tính công khai

argmax

::tensorflow::Output argmax

hoạt động

Operation operation

đầu ra

::tensorflow::Output output

Chức năng công cộng

MaxPoolWithArgmax

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

MaxPoolWithArgmax

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

Các hàm tĩnh công khai

Bao gồmBatchInIndex

Attrs IncludeBatchInIndex(
  bool x
)

Targmax

Attrs Targmax(
  DataType x
)