tf.tile
时间: 2023-06-22 15:17:44 浏览: 172
`tf.tile` 是 TensorFlow 的一个函数,用于在指定的维度上复制张量的数据。它的函数原型如下:
```python
tf.tile(
input, multiples, name=None
)
```
其中,
- `input`:需要复制的张量,可以是任意维度。
- `multiples`:一个由整数构成的张量,表示在每个维度上复制的倍数。`multiples` 的形状必须和 `input` 的形状相同或者是 `input` 形状的前缀。例如,当 `input` 的形状为 `(2, 3, 4)` 时,`multiples` 可以是 `(1, 2, 2)` 或 `(2, 1, 1)` 或 `(2, 3, 1, 1)` 等形状。
- `name`:操作的名称。
例如,假设 `x` 是一个形状为 `(3, 2)` 的张量,我们可以使用 `tf.tile` 将其在第一个维度上复制两次,第二个维度上复制三次,代码如下:
```python
import tensorflow as tf
x = tf.constant([[1, 2], [3, 4], [5, 6]])
x_tile = tf.tile(x, multiples=[2, 3])
print(x_tile)
```
执行结果如下:
```
tf.Tensor(
[[1 2 1 2 1 2]
[3 4 3 4 3 4]
[5 6 5 6 5 6]
[1 2 1 2 1 2]
[3 4 3 4 3 4]
[5 6 5 6 5 6]], shape=(6, 6), dtype=int32)
```
上面的代码中,`x_tile` 的形状为 `(6, 6)`,其中前两行是原来的 `x`,后两行是 `x` 在第一个维度上复制的结果,后三列是 `x` 在第二个维度上复制的结果。
阅读全文