【发布时间】：2021-06-28 02:05:56
【问题描述】：
在 Flux.jl 中，我想将一层中的一些数据转换为一维数组，以便将其输入到下一层。有没有内置函数可以做到这一点？
【问题讨论】：
【发布时间】：2021-06-28 02:05:56
【问题描述】：
Flux 提供了一个内置的Flux.flatten() 函数，according to the Flux docs:
将任意形状的输入重塑为矩阵形状的输出，保留最后一维的大小。
您可以看到如下示例：
julia> rand(Int8, 3,4,2) # batch of two matrices, each 3 x 4
3×4×2 Array{Int8, 3}:
[:, :, 1] =
-29 -97 61 1
34 -53 4 -16
74 46 41 -40
[:, :, 2] =
10 25 -86 77
94 -48 114 1
-68 -125 35 68
julia> Flux.flatten(ans) # batch of two vectors
12×2 Matrix{Int8}:
-29 10
34 94
74 -68
-97 25
-53 -48
46 -125
61 -86
4 114
41 35
1 77
-16 1
-40 68
【讨论】：