pytorch改变tensor维度的方法，包括 view、unsqueeze、squeeze、transpose等方法
2023-01-02 17:26:11
60KB
1
pytorch改变tensor维度
Approach 1: add dimension with None
Approach 2: unsqueeze
Approach 3: view
Approach 4: reshape
Approach 5: resize_
如果只想添加一个维度（例如，为批次添加第0个维度），则使用unsqueze（0）。如果要完全改变维度，请使用reshape（）。
2022-11-25 15:27:11
139KB
1