本文简要介绍python语言中 torch.vsplit
的用法。
用法:
torch.vsplit(input, indices_or_sections) → List of Tensors
input(Tensor) -张量分裂。
indices_or_sections(Tensor,int或者list或者python的元组:ints) -请参阅
torch.tensor_split()
中的参数。
根据
indices_or_sections
将具有两个或多个维度的张量input
垂直拆分为多个张量。每个拆分都是input
的一个视图。这等效于调用 torch.tensor_split(input, indices_or_sections, dim=0) (拆分维度为 0),除非
indices_or_sections
是整数,它必须将拆分维度均分,否则运行时错误将被抛出。该函数基于 NumPy 的
numpy.vsplit()
。>>> t = torch.arange(16.0).reshape(4,4) >>> t tensor([[ 0., 1., 2., 3.], [ 4., 5., 6., 7.], [ 8., 9., 10., 11.], [12., 13., 14., 15.]]) >>> torch.vsplit(t, 2) (tensor([[0., 1., 2., 3.], [4., 5., 6., 7.]]), tensor([[ 8., 9., 10., 11.], [12., 13., 14., 15.]])) >>> torch.vsplit(t, [3, 6]) (tensor([[ 0., 1., 2., 3.], [ 4., 5., 6., 7.], [ 8., 9., 10., 11.]]), tensor([[12., 13., 14., 15.]]), tensor([], size=(0, 4)))
例子:
参数:
相关用法
- Python PyTorch vstack用法及代码示例
- Python PyTorch vdot用法及代码示例
- Python PyTorch view_as_real用法及代码示例
- Python PyTorch vocab用法及代码示例
- Python PyTorch view_as_complex用法及代码示例
- Python PyTorch vjp用法及代码示例
- Python PyTorch vander用法及代码示例
- Python PyTorch var_mean用法及代码示例
- Python PyTorch verify_skippables用法及代码示例
- Python PyTorch vector_norm用法及代码示例
- Python PyTorch vhp用法及代码示例
- Python PyTorch var用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
- Python PyTorch cholesky用法及代码示例
- Python PyTorch ELU用法及代码示例
- Python PyTorch ScaledDotProduct.__init__用法及代码示例
- Python PyTorch gumbel_softmax用法及代码示例
- Python PyTorch get_tokenizer用法及代码示例
- Python PyTorch saved_tensors_hooks用法及代码示例
- Python PyTorch positive用法及代码示例
- Python PyTorch renorm用法及代码示例
- Python PyTorch AvgPool2d用法及代码示例
- Python PyTorch MaxUnpool3d用法及代码示例
- Python PyTorch Bernoulli用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.vsplit。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。