all AI news
flatten(), unflatten() and ravel() in PyTorch
April 22, 2024, 11:41 p.m. | Super Kai (Kazuya Ito)
DEV Community dev.to
flatten() can remove zero or more dimensions from a 0D or more D tensor as shown below:
*Memos:
flatten() can be used with torch and a tensor.- The tensor of zero or more integers, floating-point numbers, complex numbers or boolean values can be used.
- The 2nd argument(int) with torch or the 1st argument(int) with a tensor is start_dim(Optional-Default:0) which is the 1st dimension.
- The 3rd argument(int) with torch or the 2nd argument(int) with a tensor is end_dim(Optional-Default:-1) which is the …
dimensions flatten integers memos numbers pytorch tensor torch unflatten values
More from dev.to / DEV Community
Jobs in AI, ML, Big Data
Artificial Intelligence – Bioinformatic Expert
@ University of Texas Medical Branch | Galveston, TX
Lead Developer (AI)
@ Cere Network | San Francisco, US
Research Engineer
@ Allora Labs | Remote
Ecosystem Manager
@ Allora Labs | Remote
Founding AI Engineer, Agents
@ Occam AI | New York
AI Engineer Intern, Agents
@ Occam AI | US