[B, T, C, H, W]
T = 30
H = W = 224
C = 3
[0.485, 0.456, 0.406]
[0.229, 0.224, 0.225]
float32
bfloat16
-