maxframe.tensor.vstack

maxframe.tensor.vstack(tup)

沿垂直方向（行方向）顺序堆叠张量。

这相当于在形状为 (N,) 的一维张量被重塑为 (1,N) 后，沿第一个轴进行连接操作。该函数可以重建由 vsplit 分割的张量。

此函数对于最多三维的张量最有意义。例如，对于具有高度（第一轴）、宽度（第二轴）和 r/g/b 通道（第三轴）的像素数据。函数 concatenate、stack 和 block 提供了更通用的堆叠和连接操作。

参数:

tup (sequence of tensors) -- 除了第一个轴之外，张量必须在所有其他轴上具有相同的形状。一维张量必须具有相同的长度。

返回:

stacked -- 通过堆叠给定张量形成的张量，至少为二维。

返回类型:

Tensor

参见

stack

沿新轴连接一系列张量。

concatenate

沿现有轴连接一系列张量。

示例

>>> import maxframe.tensor as mt

>>> a = mt.array([1, 2, 3])
>>> b = mt.array([2, 3, 4])
>>> mt.vstack((a,b)).execute()
array([[1, 2, 3],
       [2, 3, 4]])

>>> a = mt.array([[1], [2], [3]])
>>> b = mt.array([[2], [3], [4]])
>>> mt.vstack((a,b)).execute()
array([[1],
       [2],
       [3],
       [2],
       [3],
       [4]])