import tensorflow as tf
if __name__ == "__main__":
weights = tf.get_variable(
name="weights",
shape=[9, 3, 1, 64],
dtype=tf.float32,
initializer=tf.truncated_normal_initializer()
),
print(tf.shape(weights, name="weights_shape"))
Output:
Tensor("weights_shape:0", shape=(5,), dtype=int32)
Can't figure out why the shape of "weights" does not match the given shape.
Any help would be appreciated!