cut a 1 dimensional tensor via tf.math.argmin

后端 未结 0 773
悲哀的现实
悲哀的现实 2021-01-13 08:26

I want to cut a tensor of shape (None,) off at the position of it\'s smallest value. Apperently

cut = tf.math.argmin(x,axis = 0)
x = tf.slice(x,[0],[cut])


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题