Commit 828f2ba3 authored by Boris Bonev's avatar Boris Bonev
Browse files

removing some more docstrings

parent 61dd6cf1
...@@ -503,14 +503,6 @@ class DiscreteContinuousConvS2(DiscreteContinuousConv): ...@@ -503,14 +503,6 @@ class DiscreteContinuousConvS2(DiscreteContinuousConv):
@property @property
def psi_idx(self): def psi_idx(self):
"""
Get the convolution tensor index
Returns
-------
psi_idx: torch.Tensor
Convolution tensor index
"""
return torch.stack([self.psi_ker_idx, self.psi_row_idx, self.psi_col_idx], dim=0).contiguous() return torch.stack([self.psi_ker_idx, self.psi_row_idx, self.psi_col_idx], dim=0).contiguous()
def forward(self, x: torch.Tensor) -> torch.Tensor: def forward(self, x: torch.Tensor) -> torch.Tensor:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment