Unstacks a tf.Tensor of rank-R into a list of rank-(R-1) tf.Tensors.
tf.Tensor
R
(R-1)
const a = tf.tensor2d([1, 2, 3, 4], [2, 2]);tf.unstack(a).forEach(tensor => tensor.print()); Copy
const a = tf.tensor2d([1, 2, 3, 4], [2, 2]);tf.unstack(a).forEach(tensor => tensor.print());
A tensor object.
Optional
The axis to unstack along. Defaults to 0 (the first dim).
Unstacks a
tf.Tensor
of rank-R
into a list of rank-(R-1)
tf.Tensor
s.