torchvision.transforms.Pad和torchvision.transforms.functional.pad的区别

from torchvision import transforms as T
from torchvision.transforms import functional as F

T.Pad
F.pad

在学习FCN语义分割的代码时,遇到了F.pad。当时纳闷,其与T.Pad的区别,然后看了下torchvision.transforms的源码,根据个人的理解,现在说说区别如下:

T.Pad是一个类,该类里面的forward函数里面返回的就是F.pad。

F.pad是一个函数,可以赋值给另外一个变量。

你可能感兴趣的:(Pytorch,pytorch,transforms)