[python] Tensorflow tf.math.argmax 함수
Tensorflow에서 제공하는 tf.math.argmax함수는 tensor의 가장 큰 값의 index를 return해주는 함수이다. 기본 형태는 tf.math.argmax(input, axis = None, output_type = tf.dtype.int64, name = None) 위와 같다. input값은 tensor여야 하고, axis는 default는 0이고, max값을 찾을 축을 의미한다. 예제 1) a = tf.constant([2, 20, 30, 6, 3]) tf.math.argmax(a) a에서는 30이 가장 큰 값이고, 30의 index는 2 이므로, argmax 함수의 return 값은 2가 된다. 예제 2) a = tf.constant([[2, 20, 30, 3, 6], [3, 11..
이것저것 자료/Python
2021. 8. 4. 13:15
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- LGSVL
- numpy
- CUDA
- PIP
- 파이썬
- dfs
- 동적프로그래밍
- 코딩테스트
- 설치하기
- Python
- 다익스트라
- n과m
- BFS
- notfound
- 설치
- tensorflow
- shellscript
- docker
- 백트래킹
- 카카오
- 프로그래머스
- torchscript
- pytorch
- torch
- 이것이코딩테스트다
- 백준
- 최소신장트리
- error
- version
- matplotlib
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 1 | 2 | 3 | ||||
| 4 | 5 | 6 | 7 | 8 | 9 | 10 |
| 11 | 12 | 13 | 14 | 15 | 16 | 17 |
| 18 | 19 | 20 | 21 | 22 | 23 | 24 |
| 25 | 26 | 27 | 28 | 29 | 30 | 31 |
글 보관함