Можете ли вы передать переменную члену данных в Python, как это сделано в Pytorch?

Question

Вот пример кода из nn.Module документации по pytorch:

class Model(nn.Module):
    def __init__(self):
        super(Model, self).__init__()
        self.conv1 = nn.Conv2d(1, 20, 5)
        self.conv2 = nn.Conv2d(20, 20, 5)

    def forward(self, x):
       x = F.relu(self.conv1(x))
       return F.relu(self.conv2(x))

Здесь мы передаем x в self.conv1() в функции forward.Однако self.conv1 - это переменная, которая может быть из этой строки self.conv1 = nn.Conv2d(1, 20, 5) в функции __init__.

Как это возможно?

Answer 1

Передача всего, что находится в pytorch, слое или сети, выполняется функционально var(x).Это сделано в Python с перезаписью __call__() встроенным.Попробуй сам.Создайте класс, перезапишите __call__() и используйте его как функцию.

Answer 2

Я вижу, это хороший вопрос.

self.conv1 = nn.Conv2d(1, 20, 5)

self.conv1 действительно переменная, но она также является специальной переменной класса, полученной nn.Module (читай: модуль).

И это также вызываемый, реализованный __call__ метод, и поэтому мы можем его вызвать.