typeTensor()不能是types或string

2024-04-26 12:44:21 发布

您现在位置:Python中文网/ 问答频道 /正文

我得到一个张量作为参数,现在我想创建一些变量,我正在尝试tf.get_变量(我不想用tf.变量)在

input=tensor_argument


sequence_length = tf.shape(inputs)[1]  # the length 

hidden_size = tf.shape(inputs)[2] # hidden size 

W_omega = tf.get_variable(name='w_omega',shape=[sequence_length,sequence_length],dtype=tf.float32,initializer=tf.random_uniform_initializer(-0.01,0.01))

当我运行这个代码时,我得到了这个错误

^{pr2}$

我试过了tf.形状以及tf.尺寸()但什么也没用

示例:

import tensorflow as tf
import numpy as np


data=np.random.randint(0,10,[2,4,300])

tensor_va=tf.constant(data)

d=tf.shape(tensor_va)[1]
W_omega = tf.get_variable(name='a_omega', shape=[d,d], dtype=tf.float32,
                          initializer=tf.random_uniform_initializer(-0.01, 0.01))

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(W_omega))

如何使用张量或将张量转换为整型变量?在


Tags: sizegettfasrandomvariablelengthhidden