torch.stack accepts axis keyword argument even though it’s not in the definition
torch.stack documentation shows the 2nd argument is dim which is a keyword argument. However, I noticed torch.stack also accepts axis instead of dim.
torch.stack documentation shows the 2nd argument is dim which is a keyword argument. However, I noticed torch.stack also accepts axis instead of dim.