TensorFlow 1 version
|
View source on GitHub
|
Creates a tensor filled with a scalar value.
tf.fill(
dims, value, name=None
)
This operation creates a tensor of shape dims and fills it with value.
For example:
# Output tensor has shape [2, 3].
fill([2, 3], 9) ==> [[9, 9, 9]
[9, 9, 9]]
tf.fill differs from tf.constant in a few ways:
tf.fillonly supports scalar contents, whereastf.constantsupports Tensor values.tf.fillcreates an Op in the computation graph that constructs the actual Tensor value at runtime. This is in contrast totf.constantwhich embeds the entire Tensor into the graph with aConstnode.- Because
tf.fillevaluates at graph runtime, it supports dynamic shapes based on other runtime Tensors, unliketf.constant.
Args | |
|---|---|
dims
|
A Tensor. Must be one of the following types: int32, int64. 1-D.
Represents the shape of the output tensor.
|
value
|
A Tensor. 0-D (scalar). Value to fill the returned tensor.
@compatibility(numpy) Equivalent to np.full @end_compatibility
|
name
|
A name for the operation (optional). |
Returns | |
|---|---|
A Tensor. Has the same type as value.
|
TensorFlow 1 version
View source on GitHub