在TensorFlow中,可以使用tf.reshape()
函數來改變張量的形狀。以下是檢查和改變張量形狀的示例代碼:
import tensorflow as tf
# 創建一個張量
tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
# 打印張量的形狀
print("原始張量的形狀:", tensor.shape)
# 改變張量的形狀為(3, 2)
reshaped_tensor = tf.reshape(tensor, (3, 2))
# 打印改變后張量的形狀
print("改變后張量的形狀:", reshaped_tensor.shape)
在上面的示例中,我們首先創建了一個形狀為(2, 3)的張量,并打印了它的形狀。然后使用tf.reshape()
函數將該張量的形狀改變為(3, 2),并打印了改變后張量的形狀。