def _create_zero_arrays(size):
    size = _state_size_with_prefix(size, prefix=[batch_size])
    return array_ops.zeros(array_ops.stack(size), _infer_state_dtype(dtype, state))
def _create_zero_arrays(size):
    size = _state_size_with_prefix(size, prefix=[batch_size])
    return array_ops.zeros(array_ops.stack(size), _infer_state_dtype(dtype, state))