tf.contrib.kfac.loss_functions.insert_slice_in_zeros( slice_to_insert, dim, dim_size, position )
Defined in tensorflow/contrib/kfac/python/ops/loss_functions.py
.
Inserts slice into a larger tensor of zeros.
Forms a new tensor which is the same shape as slice_to_insert, except that the dimension given by 'dim' is expanded to the size given by 'dim_size'. 'position' determines the position (index) at which to insert the slice within that dimension.
Assumes slice_to_insert.shape[dim] = 1.
slice_to_insert
: The slice to insert.dim
: The dimension which to expand with zeros.dim_size
: The new size of the 'dim' dimension.position
: The position of 'slice_to_insert' in the new tensor.The new tensor.
ValueError
: If the slice's shape at the given dim is not 1.
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/kfac/loss_functions/insert_slice_in_zeros