torch.nn.functional.conv_transpose3d#
- torch.nn.functional.conv_transpose3d(input,weight,bias=None,stride=1,padding=0,output_padding=0,groups=1,dilation=1)→Tensor#
Applies a 3D transposed convolution operator over an input imagecomposed of several input planes, sometimes also called “deconvolution”
This operator supportsTensorFloat32.
See
ConvTranspose3dfor details and output shape.Note
In some circumstances when given tensors on a CUDA device and using CuDNN, this operator may select a nondeterministic algorithm to increase performance. If this is undesirable, you can try to make the operation deterministic (potentially at a performance cost) by setting
torch.backends.cudnn.deterministic=True. SeeReproducibility for more information.- Parameters
input – input tensor of shape
weight – filters of shape
bias – optional bias of shape. Default: None
stride – the stride of the convolving kernel. Can be a single number or atuple
(sT,sH,sW). Default: 1padding –
dilation*(kernel_size-1)-paddingzero-padding will be added to bothsides of each dimension in the input. Can be a single number or a tuple(padT,padH,padW). Default: 0output_padding – additional size added to one side of each dimension in theoutput shape. Can be a single number or a tuple
(out_padT,out_padH,out_padW). Default: 0groups – split input into groups, should be divisible by thenumber of groups. Default: 1
dilation – the spacing between kernel elements. Can be a single number ora tuple(dT, dH, dW). Default: 1
Examples:
>>>inputs=torch.randn(20,16,50,10,20)>>>weights=torch.randn(16,33,3,3,3)>>>F.conv_transpose3d(inputs,weights)