本文简要介绍python语言中 torch.movedim
的用法。
用法:
torch.movedim(input, source, destination) → Tensor
将
input
在source
中位置的维度移动到destination
中的位置。input
的其他未显式移动的维度保持其原始顺序并出现在destination
中未指定的位置。例子:
>>> t = torch.randn(3,2,1) >>> t tensor([[[-0.3362], [-0.8437]], [[-0.9627], [ 0.1727]], [[ 0.5173], [-0.1398]]]) >>> torch.movedim(t, 1, 0).shape torch.Size([2, 3, 1]) >>> torch.movedim(t, 1, 0) tensor([[[-0.3362], [-0.9627], [ 0.5173]], [[-0.8437], [ 0.1727], [-0.1398]]]) >>> torch.movedim(t, (1, 2), (0, 1)).shape torch.Size([2, 1, 3]) >>> torch.movedim(t, (1, 2), (0, 1)) tensor([[[-0.3362, -0.9627, 0.5173]], [[-0.8437, 0.1727, -0.1398]]])
参数:
相关用法
- Python PyTorch moveaxis用法及代码示例
- Python PyTorch movielens_25m用法及代码示例
- Python PyTorch movielens_20m用法及代码示例
- Python PyTorch monitored_barrier用法及代码示例
- Python PyTorch mode用法及代码示例
- Python PyTorch mean用法及代码示例
- Python PyTorch multinomial用法及代码示例
- Python PyTorch meshgrid用法及代码示例
- Python PyTorch matrix_rank用法及代码示例
- Python PyTorch mm用法及代码示例
- Python PyTorch mv用法及代码示例
- Python PyTorch min用法及代码示例
- Python PyTorch max用法及代码示例
- Python PyTorch msort用法及代码示例
- Python PyTorch matrix_exp用法及代码示例
- Python PyTorch matmul用法及代码示例
- Python PyTorch matrix_power用法及代码示例
- Python PyTorch maximum用法及代码示例
- Python PyTorch masked_select用法及代码示例
- Python PyTorch maskrcnn_resnet50_fpn用法及代码示例
- Python PyTorch minimum用法及代码示例
- Python PyTorch multi_dot用法及代码示例
- Python PyTorch mul用法及代码示例
- Python PyTorch matrix_norm用法及代码示例
- Python PyTorch multigammaln用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.movedim。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。