tensorflow 生成指定大小的赋值0的张量 np.zeros 在TF中对应的语句 生成全0张量

2022-09-02 20:43:23 浏览数 (1)

语句结构:

代码语言:javascript复制
tf.zeros(

shape,

dtype=tf.float32,

name=None

)

举例:

代码语言:javascript复制
tf.zeros([3, 4], tf.int32)

最主要的是,shape可以接收1D张量。所以你可以

代码语言:javascript复制
tfshape = tf.shape(tensor1)

tensor2 = tf.zeros(shape=tfshape,

dtype = np.float32)

0 人点赞