def METHOD_NAME(self, requires_grad: bool = False) -> Sequence[torch.FloatTensor]:  # noqa: D102
    # same size tensors
    return (
        rand(self.batch_size, 12, generator=self.generator, device=self.device).requires_grad_(requires_grad),
        rand(self.batch_size, 12, generator=self.generator, device=self.device).requires_grad_(requires_grad),
    )