本文简要介绍python语言中 torch.gather
的用法。
用法:
torch.gather(input, dim, index, *, sparse_grad=False, out=None) → Tensor
沿
dim
指定的轴收集值。对于 3-D 张量,输出由下式指定:
out[i][j][k] = input[index[i][j][k]][j][k] # if dim == 0 out[i][j][k] = input[i][index[i][j][k]][k] # if dim == 1 out[i][j][k] = input[i][j][index[i][j][k]] # if dim == 2
input
和index
必须具有相同的维数。对于所有维度d != dim
,还需要index.size(d) <= input.size(d)
。out
将具有与index
相同的形状。请注意,input
和index
不会相互广播。例子:
>>> t = torch.tensor([[1, 2], [3, 4]]) >>> torch.gather(t, 1, torch.tensor([[0, 0], [1, 0]])) tensor([[ 1, 1], [ 4, 3]])
参数:
关键字参数:
相关用法
- Python PyTorch gather_object用法及代码示例
- Python PyTorch gammainc用法及代码示例
- Python PyTorch gammaincc用法及代码示例
- Python PyTorch gammaln用法及代码示例
- Python PyTorch gumbel_softmax用法及代码示例
- Python PyTorch get_tokenizer用法及代码示例
- Python PyTorch gradient用法及代码示例
- Python PyTorch global_unstructured用法及代码示例
- Python PyTorch greedy_partition用法及代码示例
- Python PyTorch get_gradients用法及代码示例
- Python PyTorch get_ignored_functions用法及代码示例
- Python PyTorch get_default_dtype用法及代码示例
- Python PyTorch gt用法及代码示例
- Python PyTorch gcd用法及代码示例
- Python PyTorch get_graph_node_names用法及代码示例
- Python PyTorch get_testing_overrides用法及代码示例
- Python PyTorch generate_sp_model用法及代码示例
- Python PyTorch ge用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
- Python PyTorch cholesky用法及代码示例
- Python PyTorch vdot用法及代码示例
- Python PyTorch ELU用法及代码示例
- Python PyTorch ScaledDotProduct.__init__用法及代码示例
- Python PyTorch saved_tensors_hooks用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.gather。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。