Add fill_constant_batch_size_like op
Created by: lijiancheng0614
This function creates a tensor of specified shape, dtype and batch size, and initializes this with a constant supplied in value. The batch size is obtained from the input tensor.
Attributes
- input (Variable) Tensor whose input_dim_idx’th dimension specifies the batch_size.
- shape (INTS) The shape of the output.
- dtype (INT) It could be numpy.dtype. Output data type. Default is float32.
- value (FLOAT) default 0. The value to be filled.
- input_dim_idx (INT) default 0. The index of input’s batch size dimension.
- output_dim_idx (INT) default 0. The index of output’s batch size dimension.
Outputs
- Out Tensor of specified shape will be filled with the specified value.