我想创建一个值为0.9且具有特定形状的张量。在TensorFlow中有一个这样的命令:
tf.ones_like()
我可以指定形状来创建一个全为1的向量。我想做同样的事情,但使用其他值,比如0.9,所以我想像使用tf.ones_like那样指定形状,但我想让我的张量值为0.9。
我该怎么做呢?
回答:
你可以使用fill函数
# 输出张量的形状为[2, 3]。fill([2, 3], 0.9) ==> [[0.9, 0.9, 0.9] [0.9, 0.9, 0.9]]