View source on GitHub |
Initializer which concats other intializers.
tfp.layers.BlockwiseInitializer(
initializers, sizes, validate_args=False
)
Args | |
---|---|
initializers
|
list of Keras initializers, e.g., "glorot_uniform" or
tf.keras.initializers.Constant(0.5413) .
|
sizes
|
list of int scalars representing the number of elements
associated with each initializer in initializers .
|
validate_args
|
Python bool indicating we should do (possibly expensive)
graph-time assertions, if necessary.
|
Attributes | |
---|---|
initializers
|
|
sizes
|
|
validate_args
|
Methods
from_config
@classmethod
from_config( config )
Instantiates an initializer from a configuration dictionary.
get_config
get_config()
Returns initializer configuration as a JSON-serializable dict.
__call__
__call__(
shape, dtype=None
)
Returns a tensor object initialized as specified by the initializer.
Args | |
---|---|
shape
|
Shape of the tensor. |
dtype
|
Optional dtype of the tensor. If not provided will return tensor
of tf.float32 .
|