当前位置: 代码迷 >> 综合 >> Tensorflow ——tf.slice和tf.gather
  详细解决方案

Tensorflow ——tf.slice和tf.gather

热度:9   发布时间:2024-02-02 15:35:57.0

Tensorflow ——tf.slice和tf.gather

  • tf.gather 和 tf.slice都可以根据索引寻找相应维度的子集。
    • tf.slice(input,begin,size,name=None) 按照指定的下标范围抽取连续区域的子集,每个维度一一对应。
    • tf.gather(input,begin,size.name=None) 按照指定的下标集合从 axis=0 中抽取子集,适合抽取不连续区域的子集。
  相关解决方案