Tensorflow:张量的行数和列数
Tensorflow: the number of rows and columns of tensor
在tensorflow中,如何求出tensor的行数和列数?
如果有权重张量如下:
W1 = tf.get_variable("W1", 形状=[784, 512],
初始值设定项=tf.contrib.layers.xavier_initializer())
我希望知道W1的行数和列数,并自动保存在一些参数中。
例如,
W1_rows <= W1的行数
W1_columns <= W1的列数
(我知道tf.shape或get_shape()可以打印出形状,但是这种方法不能用于在某些参数中保存形状(如W1_rows,W1_columns)
W1 = tf.get_variable("W1", shape=[784, 512],
initializer=tf.contrib.layers.xavier_initializer())
W1_rows, W1_columns = W1.get_shape()
print(W1_rows, W1_columns)
784 512
有效。谢谢你,奥斯瓦尔德。
在tensorflow中,如何求出tensor的行数和列数?
如果有权重张量如下:
W1 = tf.get_variable("W1", 形状=[784, 512], 初始值设定项=tf.contrib.layers.xavier_initializer())
我希望知道W1的行数和列数,并自动保存在一些参数中。
例如, W1_rows <= W1的行数 W1_columns <= W1的列数
(我知道tf.shape或get_shape()可以打印出形状,但是这种方法不能用于在某些参数中保存形状(如W1_rows,W1_columns)
W1 = tf.get_variable("W1", shape=[784, 512],
initializer=tf.contrib.layers.xavier_initializer())
W1_rows, W1_columns = W1.get_shape()
print(W1_rows, W1_columns)
784 512
有效。谢谢你,奥斯瓦尔德。