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

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