无法理解轴的使用？

Question

# 'x' is [1, 4]
# 'y' is [2, 5]
# 'z' is [3, 6]
stack([x, y, z])  # => [[1, 4], [2, 5], [3, 6]] (Pack along first dim.)
stack([x, y, z], axis=1)  # => [[1, 2, 3], [4, 5, 6]]

此代码中'轴'的作用是什么？

Answer 1

看看TensorFlow's stack function。 Axis告诉你堆积矢量的维度。

如果将x，y和z更改为a，b和c，可能更容易考虑。 axis = 0表示使用a，b和c的第一个元素来创建结果的第一个维度的元素。 axis = 1表示使用a，b和c的第一个元素来创建结果的第二个维度的元素。