Pads images such that dimensions are divisible by 8

Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).

Normlize image by given mean and std.

Prepare sample for usage as network input.

Residual convolution module.

Resize sample to given size (width, height).

Feature fusion block.

Normalize a tensor by given mean and std.

Normalize a tensor by given mean and std.

Height and width need to be divided by 16

Normalize a tensor by given mean and std.

2D image to patch embedding: (B,C,H,W) -> (B,N,D) Args: img_size: Image size. patch_size: Patch token size. in_chans