all AI news
min(), max(), argmin() and argmax() in PyTorch
April 18, 2024, 4:09 a.m. | Super Kai (Kazuya Ito)
DEV Community dev.to
*Memos:
My post explains aminmax(), amin() and amax().
My post explains minimum(), maximum(), kthvalue() and topk().
min() can get the one or more minimum values of a 0D or more D tensor as shown below:
*Memos:
min() can be called both from torch and a tensor.- Setting a dimension(dim) to the 2nd argument with torch or the 1st argument with a tensor gets zero or more 1st minimum values and the indices of them.
- Only …
More from dev.to / DEV Community
Jobs in AI, ML, Big Data
Data Architect
@ University of Texas at Austin | Austin, TX
Data ETL Engineer
@ University of Texas at Austin | Austin, TX
Lead GNSS Data Scientist
@ Lurra Systems | Melbourne
Senior Machine Learning Engineer (MLOps)
@ Promaton | Remote, Europe
Data Scientist
@ Publicis Groupe | New York City, United States
Bigdata Cloud Developer - Spark - Assistant Manager
@ State Street | Hyderabad, India