在使用时需要给d1
d2赋值,以给出调换的位置。..., h, w]=>[b, w, h, c]后,再将后面三个维度连在一起来理解,再展开成[b, c, w, h]
print(b)
此时输出会报错
RuntimeError: invalid argument...b,w,h,c]=>[b,w,h,c]=>[b,c,h,w]
print('c=', c.shape)
输出
c= torch.Size([4, 3, 28, 28])
以上两种方法虽然输出均为一致,但为验证有没有数据污染...由此看出b虽然各数据维度与a相同,但已造成了数据污染,而c没有数据污染。...下面介绍一种更加方便的转置API: permute
与transpose每次只能两两交换不同的是,permute可以一次性给出四个维度上的位置。