Class Conv2D

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Derived Type

Class Documentation

class Conv2D : public mindspore::ops::BaseOperator

2D convolution layer. Refer to Python API mindspore.ops.Conv2D for more details.

Subclassed by mindspore::ops::Conv2DFusion

Public Functions

inline Conv2D()

Constructor.

void Init(int64_t out_channel, const std::vector<int64_t> &kernel_size, int64_t mode = 1, const PadMode &pad_mode = VALID, const std::vector<int64_t> &pad = {0, 0, 0, 0}, const std::vector<int64_t> &stride = {1, 1, 1, 1}, const std::vector<int64_t> &dilation = {1, 1, 1, 1}, int64_t group = 1, const Format &format = NCHW)

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

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

Set kernel_size.

void set_stride(const std::vector<int64_t> &stride)

Set stride.

void set_dilation(const std::vector<int64_t> &dilation)

Set dilation.

void set_pad_mode(const PadMode &pad_mode)

Set pad_mode.

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

Set pad.

void set_mode(int64_t mode)

Set mode.

void set_group(int64_t group)

Set group.

void set_out_channel(int64_t out_channel)

Set out_channel.

void set_format(const Format &format)

Set format.

std::vector<int64_t> get_kernel_size() const

Get kernel_size.

Returns

kernel_size.

std::vector<int64_t> get_stride() const

Get stride.

Returns

stride.

std::vector<int64_t> get_dilation() const

Get dilation.

Returns

dilation.

PadMode get_pad_mode() const

Get pad_mode.

Returns

pad_mode.

std::vector<int64_t> get_pad() const

Get pad.

Returns

pad.

int64_t get_mode() const

Get mode.

Returns

mode.

int64_t get_group() const

Get group.

Returns

group.

int64_t get_out_channel() const

Get out_channel.

Returns

out_channel.

Format get_format() const

Get format.

Returns

format.