TensorFlow C++
TensorFlow C++
  • 请登录查看
字典 我要纠错

tensorflow::ops::Unstack

tensorflow::ops::Unstack

#include <array_ops.h>

Unpacks a given dimension of a rank-R tensor into num rank-(R-1) tensors.

Summary

Unpacks num tensors from value by chipping it along the axis dimension. For example, given a tensor of shape (A, B, C, D);

If axis == 0 then the i'th tensor in output is the slice value[i, :, :, :] and each tensor in output will have shape (B, C, D). (Note that the dimension unpacked along is gone, unlike split).

If axis == 1 then the i'th tensor in output is the slice value[:, i, :, :] and each tensor in output will have shape (A, C, D). Etc.

This is the opposite of pack.

Arguments: