unsqueeze() / squeeze()

Tensor Basics

Add or remove dimensions of size 1 from a tensor.

Python
Shift+Enter to run
1
2
3
4
5
6
7
8
9
Step 0 / 0
Speed

Write Python code and press Shift+Enter or click "Run" to visualize.

Run the operation to see step-by-step explanation.