Understand the basic 6 functions in PyTorch

squeeze(), unsqueeze(), tensor[None],max(),argmax(), and view()

Annie Wang

--

Pic from github

When I started to learn PyTorch, I found that there are various functions which seem vague to understand for me. Today I would like to summarize them with examples, which I think helpful greatly.

--

--