Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import tensorflow as tf
- import numpy as np
- a = np.array([1, 3, 7, 1, 2, 6, 0, 1])
- peaks = a[1:-1][np.diff(np.diff(a)) < 0]
- peak_values = tf.Variable(np.nonzero(np.in1d(a, peaks))[0])
- init = tf.global_variables_initializer()
- with tf.Session() as session:
- session.run(init)
- print('Peaks are at: ', session.run(peak_values))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement