TensorFlow テンソルの形状変更・確認方法【徹底解説】

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)に変更し、変更後のテンソルの形状を出力しました。