tf.reshape TensorFlow reshape 函数

tf.reshape

reshape(
    tensor,
    shape,
    name=None
)
 调整一个张量
 传入一个张量,这个操作返回一个张量,返回的张量和传入的张量有相同的值,
 被调整为传入的 shape 形状。

Reshapes a tensor.

Given tensor, this operation returns a tensor that has the same values as tensor with shape shape.

If one component of shape is the special value -1, the size of that dimension is computed so that the total size remains constant. In particular, a shape of [-1] flattens into 1-D. At most one component of shape can be -1.

If shape is 1-D or higher, then the operation returns a tensor with shape shape filled with the values of tensor. In this case, the number of elements implied by shape must be the same as the number of elements in tensor.

Related posts

Leave a Comment