paddle / unbind

unbind¶

paddle. unbind ( input, axis=0 ) [源代码] ¶

该OP将输入Tensor按照指定的维度分割成多个子Tensor。

参数¶

input (Tensor) - 输入变量，数据类型为float32，float64，int32，int64的多维Tensor。

axis (int32|int64，可选) - 数据类型为int32或int64,表示需要分割的维度。如果axis < 0，则划分的维度为rank(input) + axis。默认值为0。

返回¶

Tensor, 分割后的Tensor列表。

代码示例¶

import paddle
import numpy as np
# input is a Tensor which shape is [3, 4, 5]
np_input = np.random.rand(3, 4, 5).astype('float32')
input = paddle.to_tensor(np_input)
[x0, x1, x2] = paddle.unbind(input, axis=0)
# x0.shape [4, 5]
# x1.shape [4, 5]
# x2.shape [4, 5]
[x0, x1, x2, x3] = paddle.unbind(input, axis=1)
# x0.shape [3, 5]
# x1.shape [3, 5]
# x2.shape [3, 5]
# x3.shape [3, 5]

此页内容是否对您有帮助

PaddlePaddle

paddle

paddle.amp

paddle.callbacks

paddle.compat

paddle.distributed

paddle.distribution

paddle.fluid

paddle.io

paddle.jit

paddle.metric

paddle.nn

paddle.onnx

paddle.optimizer

paddle.regularizer

paddle.static

paddle.sysconfig

paddle.text

paddle.utils

paddle.vision

paddle / unbind

unbind¶

参数¶

返回¶

代码示例¶