当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.repeat用法及代码示例


重复 input 的元素。

用法

tf.repeat(
    input, repeats, axis=None, name=None
)

参数

  • input N 维张量。
  • repeats 一维int张量。每个元素的重复次数。广播重复以适应给定轴的形状。 len(repeats) 必须等于 input.shape[axis] 如果轴不是无。
  • axis An int. 沿其重复值的轴。默认情况下(axis=None)，使用扁平化输入数组，并返回扁平化输出数组。
  • name 操作的名称。

返回

  • 与 input 具有相同形状的张量，除了沿给定轴。如果axis为None，则输出数组被展平以匹配展平的输入数组。

另见tf.concat、tf.stack、tf.tile。

示例用法：

repeat(['a', 'b', 'c'], repeats=[3, 0, 2], axis=0)
<tf.Tensor:shape=(5,), dtype=string,
numpy=array([b'a', b'a', b'a', b'c', b'c'], dtype=object)>
repeat([[1, 2], [3, 4]], repeats=[2, 3], axis=0)
<tf.Tensor:shape=(5, 2), dtype=int32, numpy=
array([[1, 2],
       [1, 2],
       [3, 4],
       [3, 4],
       [3, 4]], dtype=int32)>
repeat([[1, 2], [3, 4]], repeats=[2, 3], axis=1)
<tf.Tensor:shape=(2, 5), dtype=int32, numpy=
array([[1, 1, 2, 2, 2],
       [3, 3, 4, 4, 4]], dtype=int32)>
repeat(3, repeats=4)
<tf.Tensor:shape=(4,), dtype=int32, numpy=array([3, 3, 3, 3], dtype=int32)>
repeat([[1,2], [3,4]], repeats=2)
<tf.Tensor:shape=(8,), dtype=int32,
numpy=array([1, 1, 2, 2, 3, 3, 4, 4], dtype=int32)>

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.repeat。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。