tf.constantconstant( value, dtype=None, shape=None, name='Const', verify_shape=F

最新推荐文章于 2022-12-12 16:05:59 发布

纶巾

最新推荐文章于 2022-12-12 16:05:59 发布

阅读量373

点赞数

分类专栏： tensorflow

本文链接：https://blog.csdn.net/WWWWWWGJ/article/details/87819000

版权

tensorflow 专栏收录该内容

14 篇文章 0 订阅

订阅专栏

constant(
    value,
    dtype=None,
    shape=None,
    name='Const',
    verify_shape=False
)

根据 value 的值生成一个 shape 维度的常量张量

参数名	必选	类型	说明
value	是	常量数值或者 list	输出张量的值
dtype	否	dtype	输出张量元素类型
shape	否	1 维整形张量或 array	输出张量的维度
name	否	string	张量名称
verify_shape	否	Boolean	检测 shape 是否和 value 的 shape 一致，若为 Fasle，不一致时，会用最后一个元素将 shape 补全

示例：

import tensorflow as tf
import numpy as np

a = tf.constant([1,2,3,4,5,6],shape=[2,3])
b = tf.constant(-1,shape=[3,2])
c = tf.matmul(a,b)

e = tf.constant(np.arange(1,13,dtype=np.int32),shape=[2,2,3])
f = tf.constant(np.arange(13,25,dtype=np.int32),shape=[2,3,2])
g = tf.matmul(e,f)

with tf.Session() as sess:
    print(sess.run(a))
    print('############################')
    print(sess.run(b))
    print('############################')
    print(sess.run(c))
    print('############################')
    print(sess.run(e))
    print('############################')
    print(sess.run(f))
    print('############################')
    print(sess.run(g))

运行结果：

[[1 2 3]
 [4 5 6]]
############################
[[-1 -1]
 [-1 -1]
 [-1 -1]]
############################
[[ -6  -6]
 [-15 -15]]
############################
[[[ 1  2  3]
  [ 4  5  6]]

 [[ 7  8  9]
  [10 11 12]]]
############################
[[[13 14]
  [15 16]
  [17 18]]

 [[19 20]
  [21 22]
  [23 24]]]
############################
[[[ 94 100]
  [229 244]]

 [[508 532]
  [697 730]]]