Flatten用法

在机器学习和深度学习中，Flatten 是一种常用的操作，其作用是将多维数组（如张量）压缩成一维数组（向量）。在 Python 中，我们可以使用 NumPy 或 TensorFlow 等库来实现 Flatten 操作。

以下是 Flatten 在 TensorFlow 中的用法示例：

import tensorflow as tf

# 定义一个 2x3x4 的张量
x = tf.constant([
    [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]],
    [[13, 14, 15, 16], [17, 18, 19, 20], [21, 22, 23, 24]]
])

# Flatten 操作
x_flat = tf.reshape(x, [-1])

# 打印结果
print(x_flat)

输出结果为：

tf.Tensor(
[ 1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24], shape=(24,), dtype=int32)

可以看到，原本的 2x3x4 张量被压缩成了一个长度为 24 的一维数组。