x.unfold/F.unfold/nn.Unfold

  • stackoverflow
  • official
    直接使用x.unfold()和nn.functional.unfold()=nn.Unfold()的结果完全不一样。但基本思想是一样的,只是操作方式截然不同。

直接使用x.unfold()

import torch
h,w = 3,3
x = torch.arange(20).float

Guess you like

Origin blog.csdn.net/u010590593/article/details/118393550