PyTorch Add Dimension: Expanding a Tensor with a Dummy Axis
Ben Cook • Posted 2017-03-09 • Last updated 2021-10-21 • CodeAdding a dimension to a tensor can be important when you’re building deep learning models.
Adding a dimension to a tensor can be important when you’re building deep learning models.
In this post we’ll classify an image with PyTorch. If you prefer to skip the prose, you can checkout the Jupyter notebook. Two interesting features of PyTorch are pythonic tensor manipulation that’s similar to numpy and dynamic computational graphs, which handle recurrent neural networks in a more natural way than static … Read more