tensorflow 1.12,tensorflow.reshape
时间: 2023-08-24 10:08:46 浏览: 145
TensorFlow常用函数总结.pdf
对于 TensorFlow 1.12 版本,可以使用 `tf.reshape` 函数来改变张量的形状。`tf.reshape` 函数的语法如下:
```python
tf.reshape(tensor, shape, name=None)
```
其中,`tensor` 是需要改变形状的张量,`shape` 是一个整型列表或元组,表示新的形状。需要注意的是,新的形状的元素数量必须与原始张量的元素数量保持一致。
下面是一个使用 `tf.reshape` 的示例:
```python
import tensorflow as tf
# 创建一个形状为 (2, 3) 的张量
x = tf.constant([[1, 2, 3], [4, 5, 6]])
# 改变张量的形状为 (3, 2)
y = tf.reshape(x, [3, 2])
with tf.Session() as sess:
print(sess.run(y))
```
这将输出:
```
[[1 2]
[3 4]
[5 6]]
```
希望这可以帮助到你!如果还有其他问题,请随时问。
阅读全文