Method forward

beginner_source/blitz/data_parallel_tutorial.py:106–111

(self, input)

        self.fc = nn.Linear(input_size, output_size)

    def forward(self, input):
        output = self.fc(input)
        print("\tIn Model: input size", input.size(),
              "output size", output.size())

        return output


######################################################################