This repository was archived by the owner on Aug 21, 2025. It is now read-only.
This repository was archived by the owner on Aug 21, 2025. It is now read-only.
[vmap] Most used nn.functional ops: 1-25 #102
Description
- softmax 10522
- relu 8572
- interpolate (upsample_nearest) 7277
- pad (Finished everything but circular, which is in Python), 5207
- log_softmax 4699
- normalize (norm, clamp_min, expand_as, div) 2338
- cross_entropy 2083
- grid_sample 1970
- one_hot 1967
- mse_loss 1920
- conv2d 1593
- dropout 1516
- softplus 1385
- sigmoid 1128
- linear 1036
- gelu 930
- avg_pool2d 899
- max_pool2d 876
- nll_loss 863
- embedding 737
- tanh 664
- leaky_relu 640
- adaptive_avg_pool2d 633
- cosine_similarity 627
- unfold 609