PyTorch Upsample() 函數實現上采樣

import torch import torch.nn as nninput = torch.arange(1, 5, dtype=torch.float32).view(1,1,2,2) print(input)m = nn.Upsample(scale_factor=2, mode='bicubic', align_corners=True)m(input)

輸出結果如下：

tensor([[[[1., 2.],[3., 4.]]]]) tensor([[[[1.0000, 1.3148, 1.6852, 2.0000],[1.6296, 1.9444, 2.3148, 2.6296],[2.3704, 2.6852, 3.0556, 3.3704],[3.0000, 3.3148, 3.6852, 4.0000]]]])

Ref：

https://pytorch.org/docs/stable/generated/torch.nn.Upsample.html

總結

以上是生活随笔為你收集整理的PyTorch Upsample() 函数实现上采样的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。