Logo Questions Linux Laravel Mysql Ubuntu Git Menu
HTML CSS JAVASCRIPT SQL PYTHON PHP BOOTSTRAP JAVA JQUERY R React Kotlin
 

Index of a maximum element in TensorFlow tensor

Tags:

tensorflow

linear-algebra

How can I get index of a maximum element of a tensor along a selected axis in TensorFlow?

like image 819
Sergey S. Avatar asked May 08 '16 20:05

Sergey S.

1 Answers

Sorry, already figured that out. It is:

tf.argmax(input, axis, name=None)

https://www.tensorflow.org/api_docs/python/tf/argmax

like image 177
Sergey S. Avatar answered Sep 29 '22 15:09

Sergey S.
Sign in to Comment

Related questions

Solve small symmetric positive definite Ax = b on GPU only
What is causing the 2x slowdown in my Cython implementation of matrix vector multiplication?
Elegant implementation of n-dimensional matrix multiplication using lists?
Eigen Linear Solver for very small square matrix
Add / substract between matrix and vector in pytorch
Efficiently computing a linear combination of data.table columns
Linear Algebra usage in programming [closed]
How to write linearly dependent column in a matrix in terms of linearly independent columns?
Efficient low-rank appoximation in MATLAB
Calculate a 2D homogeneous perspective transformation matrix from 4 points in MATLAB
Easiest way to multiply vector values together?
LAPACK SVD (Singular Value Decomposition)
Matrix multiplication in Common Lisp
Is Möller-Trumbore ray intersection the fastest?
How do I check if a simplex contains the origin?
shoot projectile (straight trajectory) at moving target in 3 dimensions
What are the pros and cons of using Matrices, Euler Angles, and or Quaternions for rotation representation? [closed]
pseudo inverse of sparse matrix in python
Compute Covariance Matrix in Java
Assumptions in Mathematica's NullSpace Command for Symbolic Matrices

Donate For Us

If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!