tf.contrib.kfac.loss_functions.insert_slice_in_zeros( slice_to_insert, dim, dim_size, position )
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.