Method
__init__
(self, spatial_dims: int, in_channels: int = 3, embed_dim: int = 48, bias: bool = False)
Source from the content-addressed store, hash-verified
| 71 | """ |
| 72 | |
| 73 | def __init__(self, spatial_dims: int, in_channels: int = 3, embed_dim: int = 48, bias: bool = False): |
| 74 | super().__init__( |
| 75 | spatial_dims=spatial_dims, |
| 76 | in_channels=in_channels, |
| 77 | out_channels=embed_dim, |
| 78 | kernel_size=3, |
| 79 | strides=1, |
| 80 | padding=1, |
| 81 | bias=bias, |
| 82 | conv_only=True, |
| 83 | ) |
| 84 | |
| 85 | def forward(self, x: torch.Tensor) -> torch.Tensor: |
| 86 | x = super().forward(x) |
Callers
nothing calls this directly
Tested by
no test coverage detected