ConvTranspose#
ConvTranspose operation performs the same computation as in ConvolutionBackwardData, except the source and destination are swapped.
Operation Attributes#
|
Description |
Value Type |
|
|
---|---|---|---|---|
Controls the strides the weights tensor is moved when computing convolution |
s64 |
A s64 list containing positive values |
Required |
|
Controls number of zeros to be add to the front/top/left of spatial dimensions |
s64 |
A s64 list containing non-negative values |
Required |
|
Controls number of zeros to be add to the back/bottom/right of spatial dimensions |
s64 |
A s64 list containing non-negative values |
Required |
|
Controls the amount of stretching the kernel before convolution |
s64 |
A s64 list containing positive values (>1 means dilated convolution) |
Required |
|
Controls how the padding is calculated |
string |
|
Optional |
|
Adds
additional
amount of
padding per
each
spatial
axis in
|
s64 |
A s64 list containing non-negative values, all zeros by default |
Optional |
|
Controls how input channels and output channels are divided into |
s64 |
A positive
s64 value,
|
Optional |
|
Controls
how to
interpret
the shape
of |
string |
|
Optional |
|
Controls
how to
interpret
the shape
of
|
string |
|
Optional |
Execution Arguments#
The inputs and outputs must be provided according to the below index order when constructing an operation.
Inputs#
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
|
Required |
1 |
|
Required |
2 |
|
Optional |
@note The shape of weights is
\((in\_channels / groups, out\_channels, spatial\_shape)\)
for IOX
format or
\((spatial\_shape, out\_channels, in\_channels / groups)\)
for XOI
format. Both \(in\_channels\) and
\(out\_channels\) must be divisible by groups attribute.
Outputs#
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
|
Required |
Supported Data Types#
ConvTranspose operation supports the following data type combinations.
Src |
Weights |
Bias |
Dst |
---|---|---|---|
f32 |
f32 |
f32 |
f32 |
bf16 |
bf16 |
bf16 |
bf16 |
f16 |
f16 |
f16 |
f16 |