摘要:        
把维度进行压缩A*1*B就会变成A*B也可以使用torch.squeeze去压缩指定维度,如果压缩的指定维度不为1则返回原数组。 import torch import numpy as np x=np.arange(24).reshape((2,1,3,4)) x = torch.tensor(x    阅读全文
posted @ 2022-12-24 11:48
祥瑞哈哈哈
阅读(41)
评论(0)
推荐(0)
        
            
        
        
摘要:        
就是在指定维度前再插入一个新的维度。 import torch import numpy as np x=np.arange(24).reshape((2,3,4)) x = torch.tensor(x) print(x) y=x.permute((2, 0, 1)) print(y.shape)    阅读全文
posted @ 2022-12-24 08:05
祥瑞哈哈哈
阅读(55)
评论(0)
推荐(0)
        
            
        
        
摘要:        
permute进行的是置换。 permute的dim需要参数表示进行置换的维度。 import torch import numpy as np x=np.arange(24).reshape((2,3,4)) x = torch.tensor(x) print(x) y=x.permute((2,    阅读全文
posted @ 2022-12-24 07:44
祥瑞哈哈哈
阅读(193)
评论(0)
推荐(0)
        
 
                     
                    
                 
                    
                
 
         浙公网安备 33010602011771号
浙公网安备 33010602011771号