In lua for nngraph we use the extra bracket followed by nn module as follow
input = nn.Identity()()
the input variable show as nngraph.Node.
How can I will get the same thing in pytorch ?
In lua when we add next module or layer the previous module or layer it take as input. like
newBlock = nn.Linear(512, 0)(input)
Please let me know some similar library in pytorch.