Method body

intermediate_source/torch_export_tutorial.py:233–234

(x, y, z)

class MapModule(torch.nn.Module):
    def forward(self, xs, y, z):
        def body(x, y, z):
            return x + y + z

        return torch_map(body, xs, y, z)