nn.Upsample的参数用法

在PyTorch中，nn.Upsample模块用于对张量进行插值，以放大或缩小其尺寸。它的参数用法如下：

* scale_factor：指定插值因子。可以使用单一数值或一个张量，以表示沿着所有维度上的插值因子。如果使用单一数值，那么它必须是一个大于1的数，用于放大张量的所有维度。如果使用张量，则可以指定不同维度的插值因子。
* mode：指定插值模式。可选的模式包括'nearest'（最近邻插值）、'bilinear'（双线性插值）和'bicubic'（双三次插值）。默认模式为'bilinear'。
* align_corners：指定是否保持插值输出和输入在角点的对齐方式。默认情况下，align_corners为True，这意味着输出和输入在角点上对齐。

下面是一个使用nn.Upsample的示例代码：

import torch
import torch.nn as nn

# 创建一个大小为(1, 2, 2, 1)的张量
input = torch.randn(1, 2, 2, 1)

# 创建一个插值层，放大张量的尺寸到(1, 4, 4, 2)
upsample = nn.Upsample(scale_factor=(2, 2), mode='bilinear', align_corners=True)

# 应用插值层得到放大后的张量
output = upsample(input)

print(output.size())  # 输出：(1, 4, 4, 2)

在上面的示例中，我们创建了一个大小为(1, 2, 2, 1)的张量，并使用nn.Upsample层将其放大到(1, 4, 4, 2)的尺寸。我们指定了插值因子为(2, 2)，模式为'bilinear'，并设置了align_corners为True以保持对齐方式。最后，我们打印了输出张量的尺寸，可以看到它已经被放大到了指定的尺寸。