PiperOrigin-RevId: 576159322 Change-Id: I29fc3df294af008f55324bc49c0f597d778e3392
This commit is contained in:
committed by
Copybara-Service
parent
612de347d7
commit
a4588cee12
+1
-1
@@ -313,7 +313,7 @@
|
||||
" \"\"\"Returns 6d spatial transform and 6d velocity for all bodies.\"\"\"\n",
|
||||
" x = Transform(pos=data.xpos[1:, :], rot=data.xquat[1:, :])\n",
|
||||
" cvel = Motion(vel=data.cvel[1:, 3:], ang=data.cvel[1:, :3])\n",
|
||||
" offset = data.xipos[1:, :] - data.subtree_com[\n",
|
||||
" offset = data.xpos[1:, :] - data.subtree_com[\n",
|
||||
" self.model.body_rootid[np.arange(1, self.model.nbody)]]\n",
|
||||
" xd = Transform.create(pos=offset).vmap().do(cvel)\n",
|
||||
" return x, xd\n"
|
||||
|
||||
Reference in New Issue
Block a user