PyTorch: Handling multiple functions and arguments

Question

I have a large number of neural networks (such that I have to use list comprehension to produce them), say [f_1, f_2, f_3 ...], collectively denoted by F.

I have the argument X = [x_1, x_2, x_3 ...], such that input tensor x_i is intended for the network f_i. The tensors I have will be big, owing to the data and then the gradients.

Is there an elegant (and efficient) way of obtaining the sequence [f_1(x_1), f_2(x_2), f_3(x_3) ...]?

flawr · Accepted Answer · 2022-07-04 11:53:59Z

2

You could again use a list comprehension:

out = [f(x) for f,x in zip(F,X)]

answered Jul 4, 2022 at 11:53

flawr

11.7k4 gold badges49 silver badges83 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

PyTorch: Handling multiple functions and arguments

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related