在TensorFlow中,tf.reshape()
函數用于更改張量的形狀,即重新設置張量的維度。其語法為:
tf.reshape(tensor, shape)
其中,tensor
為要進行形狀改變的張量,shape
為新的形狀。需要注意的是,新的形狀的總元素個數必須與原張量的總元素個數相同,否則會報錯。
示例:
import tensorflow as tf
# 定義一個張量
x = tf.constant([[1, 2],
[3, 4],
[5, 6]])
# 將張量x的形狀改變為(2, 3)
reshaped_x = tf.reshape(x, [2, 3])
# 查看改變后的張量
print(reshaped_x)
在以上示例中,tf.reshape()
函數將原來形狀為(3, 2)的張量x
重塑為新的形狀為(2, 3)的張量reshaped_x
。