Method forward

bvr/training/train_bc.py:124–133

Forward pass: observation tensor → action tensor. Args: x: Batch of observations, shape (batch_size, obs_dim). Returns: Batch of actions, shape (batch_size, act_dim), values in [-1, 1].

(self, x: torch.Tensor)

        )

    def forward(self, x: torch.Tensor) -> torch.Tensor:
        """Forward pass: observation tensor → action tensor.

        Args:
            x: Batch of observations, shape (batch_size, obs_dim).

        Returns:
            Batch of actions, shape (batch_size, act_dim), values in [-1, 1].
        """
        return self.net(x)

    def export_layers(self) -> list[dict]:
        """Export weights and biases for the Rust policy JSON format.