2021年3月2日
tensorflow中的gather、gather_nd、boolean_mask
摘要： 1. gather import tensorflow as tf a = tf.random.normal([4, 38, 8]) print(a.shape) a_1 = tf.gather(a, axis=0, indices=[3, 1]) # 第一维度的3和1，这里的indices是不用限 阅读全文
posted @ 2021-03-02 22:20 cyssmile 阅读(214) 评论(0) 推荐(0) 编辑