(name, shape, dtype, num_shards)
return concat_variable
Get a sharded variable concatenated into one tensor.
Get a sharded variable concatenated into one tensor.
Get a sharded variable concatenated into one tensor .
def _get_concat_variable(name, shape, dtype, num_shards):
    """Get a sharded variable concatenated into one tensor."""
    sharded_variable = _get_sharded_variable(name, shape, dtype, num_shards)
    if len(sharded_variable) == 1:
        return sharded_variable[0]
    
    concat_name = name + "/concat"
    concat_full_name = vs.get_variable_scope().name + "/" + concat_name + ":0"
    for value in ops.get_collection(ops.GraphKeys.CONCATENATED_VARIABLES):
        if value.name == concat_full_name:
            return value
    
    concat_variable = array_ops.concat(0, sharded_variable, name=concat_name)
    ops.add_to_collection(ops.GraphKeys.CONCATENATED_VARIABLES,
                          concat_variable)
    return concat_variable
def _get_concat_variable ( name , shape , dtype , num_shards ) : sharded_variable = _get_sharded_variable ( name , shape , dtype , num_shards ) if len ( sharded_variable ) == 1 : return sharded_variable [ 0 ] concat_name = name + "/concat" concat_full_name = vs . get_variable_scope ( ) . name + "/" + concat_name + ":0" for value in ops . get_collection ( ops . GraphKeys . CONCATENATED_VARIABLES ) : if value . name == concat_full_name : return value concat_variable = array_ops . concat ( 0 , sharded_variable , name = concat_name ) ops . add_to_collection ( ops . GraphKeys . CONCATENATED_VARIABLES , concat_variable ) return concat_variable