axis=(0,1)是啥？？？？？？

import math

import torch.nn as nn
import torch
import numpy as np

# x_ = torch.randn(1,3,127,127)
# p1 = torch.randn(1,256,31,31)
# p2 = torch.randn(1,512,15,15)
# p3 = torch.randn(1,1024,15,15)
# p4 = torch.randn(1,2048,15,15)
img = np.random.randint(3,size=(4,6,3))  # 4*6*3 == H*W*3    (720 H,1280 W ,3 C)
print(img)
print("**************")
print(img[0].shape)
# print(np.mean(img,axis=2)) # 按行求平均
print(np.mean(img,axis=1))   # 按列求平均,这种说法不对
a=np.mean(img,axis=(0,1)) #  先按照1 在按照0求平均
print("*******")
print(a)
[[[2 1 2]
  [1 1 2]
  [1 0 2]
  [1 0 1]
  [1 0 2]
  [2 1 0]]

 [[0 1 2]
  [2 0 1]
  [2 1 2]
  [0 1 1]
  [1 0 1]
  [0 2 1]]

 [[0 2 2]
  [1 2 1]
  [1 2 1]
  [2 0 2]
  [0 2 0]
  [1 2 0]]

 [[2 2 2]
  [1 2 0]
  [0 0 0]
  [2 0 1]
  [0 1 0]
  [2 0 0]]]
**************
(6, 3)
[[1.33333333 0.5        1.5       ]
 [0.83333333 0.83333333 1.33333333]
 [0.83333333 1.66666667 1.        ]
 [1.16666667 0.83333333 0.5       ]]
**************
[1.04166667 0.95833333 1.08333333]

720*1280 图片：

img[0]到img[719]个矩阵