Class Conv2DTranspose

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Derived Type

Class Documentation

class Conv2DTranspose : public mindspore::ops::BaseOperator

2D transposed convolution layer. Refer to Python API mindspore.nn.Conv2dTranspose for more details.

Subclassed by mindspore::ops::Conv2dTransposeFusion

Public Functions

inline Conv2DTranspose()

Constructor.

void Init(int64_t in_channel, 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}, const std::vector<int64_t> &dilation = {1, 1}, int64_t group = 1, const Format &format = NCHW, const std::vector<int64_t> &pad_list = {0, 0, 0, 0})

Init. Refer to the parameters of Python API mindspore.nn.Conv2dTranspose for the inputs.

void set_in_channel(int64_t in_channel)

Set in_channel.

void set_out_channel(int64_t out_channel)

Set out_channel.

virtual 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.

virtual 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_format(const Format &format)

Set format.

void set_pad_list(const std::vector<int64_t> &pad_list)

Set pad_list.

int64_t get_in_channel() const

Get in_channel.

返回

in_channel.

int64_t get_out_channel() const

Get out_channel.

返回

out_channel.

std::vector<int64_t> get_kernel_size() const

Get kernel_size.

返回

kernel_size.

std::vector<int64_t> get_stride() const

Get stride.

返回

stride.

std::vector<int64_t> get_dilation() const

Get dilation.

返回

dilation.

PadMode get_pad_mode() const

Get pad_mode.

返回

pad_mode.

std::vector<int64_t> get_pad() const

Get pad.

返回

pad.

int64_t get_mode() const

Get mode.

返回

mode.

int64_t get_group() const

Get group.

返回

group.

Format get_format() const

Get format.

返回

format.

std::vector<int64_t> get_pad_list() const

Get pad_list.

返回

pad_list.