举例flatten函数的用法
时间: 2023-09-10 17:13:49 浏览: 85
tf.layers.flatten()使用
当你有一个多维数组时,可以使用`flatten()`函数将其展平为一维数组。下面是一个例子:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
flattened_arr = arr.flatten()
print(arr)
# 输出:
# [[1 2 3]
# [4 5 6]
# [7 8 9]]
print(flattened_arr)
# 输出:
# [1 2 3 4 5 6 7 8 9]
```
在上面的例子中,我们有一个2维数组`arr`,使用`flatten()`函数将其展平为一维数组`flattened_arr`。展平后的数组就是原始数组中所有元素按照行优先顺序排列的结果。
阅读全文