Deep learning for NeuroImaging in Python.
Note
This page is a reference documentation. It only explains the function signature, and not how to use it. Please refer to the gallery for the big picture.
- surfify.models.vae.compute_output_dim(input_dim, convnet)[source]ΒΆ
Compute the output dimension of a convolutional network that takes as input a square input (H = W)
- Parameters:
input_dim : int
input height and weight
convnet : iterable[nn.Module]
iterable containing the various layers. For now, the function can only work with nn.Conv2d and IcoSpMaConv layers
- Returns:
output_dim : int
output dimension
Follow us