parameters: (name, shape, dtype, num_shards)
return_statement: return shards
docstring: Get a list of sharded variables with the given dtype.
docstring_summary: Get a list of sharded variables with the given dtype.
docstring_tokens: Get a list of sharded variables with the given dtype .
function: def _get_sharded_variable(name, shape, dtype, num_shards):
    """Get a list of sharded variables with the given dtype."""
    if num_shards > shape[0]:
      raise ValueError("Too many shards: shape=%s, num_shards=%d" %
                       (shape, num_shards))
    unit_shard_size = int(math.floor(shape[0] / num_shards))
    remaining_rows = shape[0] - unit_shard_size * num_shards
  
    shards = []
    for i in range(num_shards):
      current_size = unit_shard_size
      if i < remaining_rows:
        current_size += 1
      shards.append(vs.get_variable(name + "_%d" % i, [current_size] + shape[1:],
                                    dtype=dtype))
    return shards
function_tokens: def _get_sharded_variable ( name , shape , dtype , num_shards ) : if num_shards > shape [ 0 ] : raise ValueError ( "Too many shards: shape=%s, num_shards=%d" % ( shape , num_shards ) ) unit_shard_size = int ( math . floor ( shape [ 0 ] / num_shards ) ) remaining_rows = shape [ 0 ] - unit_shard_size * num_shards shards = [ ] for i in range ( num_shards ) : current_size = unit_shard_size if i < remaining_rows : current_size += 1 shards . append ( vs . get_variable ( name + "_%d" % i , [ current_size ] + shape [ 1 : ] , dtype = dtype ) ) return shards