tf.constant
时间: 2023-10-02 19:13:29 浏览: 86
Tensorflow tf.tile()的用法实例分析
tf.constant 是 TensorFlow 中的一个函数,用于创建一个常量张量,其值在张量的生命周期内不会发生改变。该函数的语法如下:
```python
tf.constant(
value,
dtype=None,
shape=None,
name='Const'
)
```
其中,参数 value 是指定的常量值,可以是一个标量、一个列表、一个 NumPy 数组或一个张量;dtype 是指定的数据类型,默认为 value 的数据类型;shape 是指定的张量形状,可以是一个列表或一个元组,如果不指定则根据 value 自动推断;name 是指定的张量名称,默认为 'Const'。
以下是一些示例:
```python
import tensorflow as tf
import numpy as np
# 创建一个标量常量张量
a = tf.constant(3)
# 创建一个列表常量张量
b = tf.constant([1, 2, 3, 4, 5])
# 创建一个 NumPy 数组常量张量
c = tf.constant(np.zeros((2, 3)))
# 创建一个形状为 [2, 3] 的常量张量
d = tf.constant(2, shape=(2, 3))
print(a)
print(b)
print(c)
print(d)
```
输出结果:
```
tf.Tensor(3, shape=(), dtype=int32)
tf.Tensor([1 2 3 4 5], shape=(5,), dtype=int32)
tf.Tensor(
[[0. 0. 0.]
[0. 0. 0.]], shape=(2, 3), dtype=float64)
tf.Tensor(
[[2 2 2]
[2 2 2]], shape=(2, 3), dtype=int32)
```
阅读全文