original_docstring: A helper function to expand the betas's first dim to match batch
          size such that the same beta parameters can be used for all frames in a
          video sequence.
  
          Notes:
              B: batch size
              K: number of keypoints
              D: shape dimension
  
          Args:
              batch_size: batch size
              betas: shape (B, D)
  
          Returns:
              betas_video: expanded betas
          
docstring: A helper function to expand the betas's first dim to match batch
  size such that the same beta parameters can be used for all frames in a
  video sequence.
  
  batch size
  K: number of keypoints
  D: shape dimension
docstring_tokens: A helper function to expand the betas ' s first dim to match batch size such that the same beta parameters can be used for all frames in a video sequence . batch size K : number of keypoints D : shape dimension
code: def _expand_betas(self, batch_size, betas):
          """A helper function to expand the betas's first dim to match batch
          size such that the same beta parameters can be used for all frames in a
          video sequence.
  
          Notes:
              B: batch size
              K: number of keypoints
              D: shape dimension
  
          Args:
              batch_size: batch size
              betas: shape (B, D)
  
          Returns:
              betas_video: expanded betas
          """
          # no expansion needed
          if batch_size == betas.shape[0]:
              return betas
  
          # first dim is 1
          else:
              feat_dim = betas.shape[-1]
              betas_video = betas.view(1, feat_dim).expand(batch_size, feat_dim)
  
          return betas_video
code_tokens: def _expand_betas ( self , batch_size , betas ) : if batch_size == betas . shape [ 0 ] : return betas else : feat_dim = betas . shape [ - 1 ] betas_video = betas . view ( 1 , feat_dim ) . expand ( batch_size , feat_dim ) return betas_video
short_docstring: A helper function to expand the betas's first dim to match batch
  size such that the same beta parameters can be used for all frames in a
  video sequence.
short_docstring_tokens: A helper function to expand the betas ' s first dim to match batch size such that the same beta parameters can be used for all frames in a video sequence .
comment: """A helper function to expand the betas's first dim to match batch
          size such that the same beta parameters can be used for all frames in a
          video sequence.
  
          Notes:
              B: batch size
              K: number of keypoints
              D: shape dimension
  
          Args:
              batch_size: batch size
              betas: shape (B, D)
  
          Returns:
              betas_video: expanded betas
          """$SEP$# no expansion needed$SEP$# first dim is 1