The Flux.stack function is defined here: https://github.com/FluxML/Flux.jl/blob/b78a27b01c9629099adb059a98657b995760b617/src/utils.jl#L476 , and it is very simple: stack(xs, dim) = cat(unsqueeze.(xs, dim)..., dims=dim). However, its implementation seems contrary to the “Flux Performance Tips” here:…

Why does `Flux.stack` use splatting?

baggepinnen July 19, 2021, 1:12pm 2

hcat fixes the dimension to 2, stack is generic w.r.t. the dimension.

Topic		Replies	Views
Problem with cat() General Usage	9	2710	October 2, 2018
Array manipulation New to Julia	5	1717	March 29, 2019
Most efficient way to stack vectors on the fourth dimension New to Julia	8	1542	January 21, 2019
`cat` allocates too much memory Performance question	11	1395	November 19, 2020
Stack multidimensional Array General Usage question	5	1397	July 4, 2017