Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-05-09 21:07:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Documentation fix.

Browse Source 
PiperOrigin-RevId: 341801256
This commit is contained in:
Alvaro Sanchez-Gonzalez
2020-11-11 12:01:52 +00:00
committed by Diego de Las Casas
parent 138ec089c3
commit 2d3d6cb018
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
learning_to_simulate/learned_simulator.py
+2 -2
View File
@@ -98,8 +98,8 @@ class LearnedSimulator(snt.AbstractModule):
If None, we assume all particles are the same type.
Returns:
updated_position_sequence with shape [num_particles_in_batch,
num_dimensions] for one step into the future from the input sequence.
Next position with shape [num_particles_in_batch, num_dimensions] for one
step into the future from the input sequence.
"""
input_graphs_tuple = self._encoder_preprocessor(
position_sequence, n_particles_per_example, global_context,