m0_51032217的博客

代码一： import torch a=torch.randn((2,1,3,4)) print('a=',a) b=a[:,0] print('b=',b) a= tensor([[[[-0.3767, -0.8647, -0.6132, -0.5862], [ 0.5800, 0.3467, -0.4182, -0.0916], [ 0.6316, -0.6858, -0.2674, 0.1670]]], [[[-0.1980, -