Tensorflow ——tf.slice和tf.gather

Tensorflow ——tf.slice和tf.gather

  • tf.gather 和 tf.slice都可以根据索引寻找相应维度的子集。

    • tf.slice(input,begin,size,name=None) 按照指定的下标范围抽取连续区域的子集,每个维度一一对应。
    • tf.gather(input,begin,axis=None) 按照指定的下标集合从 axis 中抽取子集,适合抽取不连续区域的子集。
  • 注意:tf.slice需要明确具体维度大小,如果在计算中出现未知大小None维,则需要使用tf.gather。tf.gather适用于整维(axis),tf.slice适用于子维。

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页