tensorflow中reshape的用法是什么

小億
121
2024-04-16 13:46:08

在TensorFlow中，tf.reshape()函數(shù)用于更改張量的形狀，即重新設(shè)置張量的維度。其語(yǔ)法為：

tf.reshape(tensor, shape)

其中，tensor為要進(jìn)行形狀改變的張量，shape為新的形狀。需要注意的是，新的形狀的總元素個(gè)數(shù)必須與原張量的總元素個(gè)數(shù)相同，否則會(huì)報(bào)錯(cuò)。

示例：

import tensorflow as tf

# 定義一個(gè)張量
x = tf.constant([[1, 2],
                 [3, 4],
                 [5, 6]])

# 將張量x的形狀改變?yōu)?2, 3)
reshaped_x = tf.reshape(x, [2, 3])

# 查看改變后的張量
print(reshaped_x)

在以上示例中，tf.reshape()函數(shù)將原來(lái)形狀為(3, 2)的張量x重塑為新的形狀為(2, 3)的張量reshaped_x。

0 踩