Infers action to take towards the given (rho, theta) based on depth vision. Args: observations (Union["TensorDict", Dict]): A dictionary containing (at least) the following: - "depth" (torch.float32): Depth image tensor (N, H, W, 1).
(
self,
observations: Union["TensorDict", Dict],
masks: Tensor,
deterministic: bool = False,
)
source not stored for this graph (policy: none)
no test coverage detected