tf.reshape
2025年02月11日
一、认识
tf.reshape(x, shape) 函数用于改变 tf.Tensor 的形状。给定一个输入张量 x，返回一个新张量，该张量的数值与输入张量相同，但形状变为 shape。特殊情况: 如果 shape 中有一个维度的值为 -1，则该维度的大小会自动计算，以确保总元素数量不变。例如，shape = [-1] 会将张量 展平（flatten） 为 1D 张量。但 shape 最多只能有一个 -1，否则会报错。形状匹配: 如果 shape 是 1D 或更高维度，则返回的张量会按照新的 shape 排列原始数据。但是，新的 shape 必须与原始张量的总元素数量相同，否则会报错。
二、语法
tf.reshape (x, shape)