code: def shift(x, offset, dim, wrap, name=None):
    
    return ShiftOperation(x, offset, dim, wrap, name=name).outputs[0]
docs: Shift operation.
  
    Shift x right by +offset in dimension dim.
  
    Args:
      x: a Tensor
      offset: an integer. If negative, shift left instead of right.
      dim: a Dimension of x
      wrap: a boolean - whether to wrap (True) or pad with zeros (False).
      name: an optional string
  
    Returns:
      a Tensor with the same shape and dtype as x