import os
import sys

import torch
import torch.nn as nn


class DepthEstimation(nn.Module):
    def __init__(self):
        super().__init__()

        self.layer_1 = nn.Linear(in_features=3, out_features=3)  

    def forward(self, x: torch.Tensor) -> torch.Tensor:
        x = self.layer_1(x)
        
        return x