Class MaxPool

• Defined in File max_pool.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Derived Type

• public mindspore::ops::MaxPoolFusion (Class MaxPoolFusion)

Class Documentation

class MaxPool : public mindspore::ops::BaseOperator

Max pooling operation. Refer to Python API mindspore.ops.MaxPool for more details.

Subclassed by mindspore::ops::MaxPoolFusion

Public Functions

inline MaxPool()

Constructor.

void Init(const std::vector<int64_t> &kernel_size = {1}, const std::vector<int64_t> &stride = {1}, const PadMode &pad_mode = VALID, const Format &format = NCHW, const std::vector<int64_t> &pad = {0, 0, 0, 0}, const RoundMode &round_mode = FLOOR)

Init. Refer to the parameters of Python API mindspore.ops.MaxPool for the inputs.

void set_pad_mode(const PadMode &pad_mode)

Set pad_mode.

void set_kernel_size(const std::vector<int64_t> &kernel_size)

Set kernel_size.

void set_strides(const std::vector<int64_t> &strides)

Set strides.

void set_format(const Format &format)

Set format.

void set_pad(const std::vector<int64_t> &pad)

Set pad.

void set_round_mode(const RoundMode &round_mode)

Set round_mode.

std::vector<int64_t> get_kernel_size() const

Get kernel_size.

返回

kernel_size.

std::vector<int64_t> get_strides() const

Get strides.

返回

strides.

PadMode get_pad_mode() const

Get pad_mode.

返回

pad_mode.

Format get_format() const

Get format.

返回

format.

std::vector<int64_t> get_pad() const

Get pad.

返回

pad.

RoundMode get_round_mode() const

Get round_mode.

返回

round_mode.