MLun Wong

icon company The Chinese University of Hong Kong icon location Hong Kong icon location ML engineer specialized in medical imaging and head and neck cancers.

Results 1 issues of MLun Wong

[Feature suggestion] Optionally return the inferred dimensions

3 comment

# Description Einops calculates automatically some of the inferred dimension. For example, in ViT, Einops is great for decomposing a tensor into stacked patches like this: ```python x = torch.ones([1,...

feature suggestion