本文简要介绍python语言中 torch.split
的用法。
用法:
torch.split(tensor, split_size_or_sections, dim=0)
将张量拆分为块。每个块都是原始张量的视图。
如果
split_size_or_sections
是整数类型,那么tensor
将被分成大小相等的块(如果可能)。如果沿给定维度dim
的张量大小不能被split_size
整除,则最后一个块将更小。如果
split_size_or_sections
是一个列表,那么tensor
将根据split_size_or_sections
被拆分为大小在dim
中的len(split_size_or_sections)
块。例子:
>>> a = torch.arange(10).reshape(5,2) >>> a tensor([[0, 1], [2, 3], [4, 5], [6, 7], [8, 9]]) >>> torch.split(a, 2) (tensor([[0, 1], [2, 3]]), tensor([[4, 5], [6, 7]]), tensor([[8, 9]])) >>> torch.split(a, [1,4]) (tensor([[0, 1]]), tensor([[2, 3], [4, 5], [6, 7], [8, 9]]))
参数:
相关用法
- Python PyTorch sparse_csr_tensor用法及代码示例
- Python PyTorch spectral_norm用法及代码示例
- Python PyTorch sparse_coo_tensor用法及代码示例
- Python PyTorch sparse_用法及代码示例
- Python PyTorch saved_tensors_hooks用法及代码示例
- Python PyTorch sqrt用法及代码示例
- Python PyTorch skippable用法及代码示例
- Python PyTorch squeeze用法及代码示例
- Python PyTorch square用法及代码示例
- Python PyTorch save_on_cpu用法及代码示例
- Python PyTorch scatter_object_list用法及代码示例
- Python PyTorch skip_init用法及代码示例
- Python PyTorch simple_space_split用法及代码示例
- Python PyTorch sum用法及代码示例
- Python PyTorch sub用法及代码示例
- Python PyTorch sentencepiece_numericalizer用法及代码示例
- Python PyTorch symeig用法及代码示例
- Python PyTorch sinh用法及代码示例
- Python PyTorch sinc用法及代码示例
- Python PyTorch std_mean用法及代码示例
- Python PyTorch slogdet用法及代码示例
- Python PyTorch symbolic_trace用法及代码示例
- Python PyTorch shutdown用法及代码示例
- Python PyTorch sgn用法及代码示例
- Python PyTorch set_flush_denormal用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.split。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。