mars.tensor.stack#
- mars.tensor.stack(tensors, axis=0, out=None)[源代码]#
Join a sequence of tensors along a new axis.
The axis parameter specifies the index of the new axis in the dimensions of the result. For example, if
axis=0
it will be the first dimension and ifaxis=-1
it will be the last dimension.- 参数
tensors (sequence of array_like) – Each tensor must have the same shape.
axis (int, optional) – The axis in the result tensor along which the input tensors are stacked.
out (Tensor, optional) – If provided, the destination to place the result. The shape must be correct, matching that of what stack would have returned if no out argument were specified.
- 返回
stacked – The stacked tensor has one more dimension than the input tensors.
- 返回类型
Tensor
参见
concatenate
Join a sequence of tensors along an existing axis.
split
Split tensor into a list of multiple sub-tensors of equal size.
block
Assemble tensors from blocks.
实际案例
>>> import mars.tensor as mt
>>> arrays = [mt.random.randn(3, 4) for _ in range(10)] >>> mt.stack(arrays, axis=0).shape (10, 3, 4)
>>> mt.stack(arrays, axis=1).shape (3, 10, 4)
>>> mt.stack(arrays, axis=2).shape (3, 4, 10)
>>> a = mt.array([1, 2, 3]) >>> b = mt.array([2, 3, 4]) >>> mt.stack((a, b)).execute() array([[1, 2, 3], [2, 3, 4]])
>>> mt.stack((a, b), axis=-1).execute() array([[1, 2], [2, 3], [3, 4]])