all AI news
chunk() and unbind() in PyTorch
April 22, 2024, 1:04 a.m. | Super Kai (Kazuya Ito)
DEV Community dev.to
*My post explains split(), hsplit() and vsplit().
chunk() can split a 1D or more D tensor into one or more tensors as shown below:
*Memos:
chunk() can be used with torch or 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 chunks(Required).
- The 3rd argument(int) with torch or the 2nd argument(int) with a tensor …
chunk integers memos numbers pytorch split tensor torch values
More from dev.to / DEV Community
Jobs in AI, ML, Big Data
Software Engineer for AI Training Data (School Specific)
@ G2i Inc | Remote
Software Engineer for AI Training Data (Python)
@ G2i Inc | Remote
Software Engineer for AI Training Data (Tier 2)
@ G2i Inc | Remote
Data Engineer
@ Lemon.io | Remote: Europe, LATAM, Canada, UK, Asia, Oceania
Artificial Intelligence – Bioinformatic Expert
@ University of Texas Medical Branch | Galveston, TX
Lead Developer (AI)
@ Cere Network | San Francisco, US