tf.unstack

代码语言：javascript复制

tf.unstack(
    value,
    num=None,
    axis=0,
    name='unstack'
)

将秩为R张量的给定维数分解为秩为(R-1)张量。通过沿着轴维对num张量进行切分，从值中解压缩num张量。如果没有指定num(默认值)，则从值的形状推断它。如果value.shape[axis]未知，将引发ValueError。

例如，给定一个形状张量(A, B, C, D);

如果axis == 0，那么输出中的第i张量就是切片值[i，:，:，:]，而输出中的每个张量都有形状(B, C, D)。(注意，与split不同的是，未打包的维度已经没有了)。如果axis == 1，则输出中的第i张量为切片值[:，i，:，:]，输出中的每个张量都有形状(A, C, D)等。这是堆栈的反面。

参数:

• value: 一个秩为R的> 0张量要被解压。
• num: 一个int类型， 一个整型数。尺寸轴的长度。如果没有(默认值)就自动推断。
• axis: 一个整型数。沿着整型数展开堆栈。默认为第一个维度。负值环绕，所以有效范围是[-R, R]。
• name: 操作的名称(可选)。

返回值:

• 张量对象的列表从值中分解。

异常:

• ValueError: If num is unspecified and cannot be inferred.
• ValueError: If axis is out of the range [-R, R).

原链接: https://tensorflow.google.cn/versions/r1.9/api_docs/python/tf/unstack?hl=en

axis int range split valueerror

1 人点赞