.. _comparison-conv-transpose3d:

=========================
ConvTranspose3d 差异对比
=========================

.. panels::

  torch.nn.ConvTranspose3d
  ^^^^^^^^^^^^^^^^^^^^^^^^^^^

  .. code-block:: python

     torch.nn.ConvTranspose3d(
        in_channels,
        out_channels,
        kernel_size,
	stride=1,
	padding=0,
        output_padding=0,
	groups=1,
	bias=True,
	dilation=1,
	padding_mode='zeros',
	device=None,
	dtype=None
     )

  更多请查看 :py:class:`torch.nn.ConvTranspose3d`.

  ---

  megengine.module.ConvTranspose3d
  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

  .. code-block:: python

     megengine.module.ConvTranspose3d(
	in_channels,
        out_channels,
        kernel_size,
	stride=1,
	padding=0,
        output_padding=0,
	dilation=1,
	groups=1,
	bias=True,
        conv_mode='cross_correlation'
	compute_mode='default'
 	** kwargs
     )

  更多请查看 :py:class:`megengine.module.ConvTranspose3d`.

使用差异
--------

padding
~~~~~~~~~~~~
   PyTorch padding 可以是单个数字或元组，MegEngine padding 仅支持数值填充 0.

compute_mode 参数
~~~~~~~~~~~~~~~~~~~
   MegEngine 中包含  ``compute_mode`` 参数，PyTorch 中无此参数，该参数用于指定计算模式，当设置 “default” 时, 不会对中间结果的精度有特殊要求。当设置 “float32” 时, “float32” 将被用作中间结果的累加器, 但是只有当输入和输出的 dtype 是 float16 时有效。 

conv_mode 参数
~~~~~~~~~~~~~~~
MegEngine 中包含  ``conv_mode`` 参数，PyTorch 中无此参数，该参数用于指定卷积模式，默认值为 “cross_correlation” 时, 表示使用交叉相关计算模式。在这种模式下，输入张量被视为滤波器的目标图像，而滤波器在输入上进行滑动以生成输出张量。


.. code-block::: python

    import megengine
    import torch
    import numpy as np
    # 定义输入张量
    input_tensor1 = torch.randn(1, 3, 128, 64, 64)
    input_tensor2 = megengine.random.normal(size=(1,3,64,64))

    # 使用MegEngine的ConvTranspose3d
    me_conv_transpose = megengine.module.ConvTranspose3d(in_channels=3, out_channels=64, kernel_size=3, stride=2, padding=1)
    me_output = me_conv_transpose(input_tensor2.astype(np.float32))

    # 使用PyTorch的ConvTranspose3d
    torch_conv_transpose = torch .nn.ConvTranspose3d(in_channels=3, out_channels=64, kernel_size=3, stride=2, padding=1)
    torch_output = torch_conv_transpose(input_tensor1)

    # 打印输出结果
    print("MegEngine output:", me_output.numpy())
    print("PyTorch output:", torch_output.numpy())