code: def vertices2joints(J_regressor, vertices):
      """Calculates the 3D joint locations from the vertices
  
      Parameters
      ----------
      J_regressor : torch.tensor JxV
          The regressor array that is used to calculate the joints from the
          position of the vertices
      vertices : torch.tensor BxVx3
          The tensor of mesh vertices
  
      Returns
      -------
      torch.tensor BxJx3
          The location of the joints
      """
  
      return torch.einsum("bik,ji->bjk", [vertices, J_regressor])
  
imports: from __future__ import absolute_import
  from __future__ import print_function
  from __future__ import division
  import os.path
  import pickle
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 6 for calling a global function with example usage: vertices2joints(J_regressor, vertices) and returns: torch