本文简要介绍python语言中 torch.take_along_dim
的用法。
用法:
torch.take_along_dim(input, indices, dim, *, out=None) → Tensor
out(Tensor,可选的) -输出张量。
沿着给定的
dim
在indices
的一维索引处从input
中选择值。沿维度返回索引的函数,如
torch.argmax()
和torch.argsort()
,旨在与此函数一起使用。请参阅下面的示例。注意
这个函数类似于 NumPy 的
take_along_axis
。另见torch.gather()
。例子:
>>> t = torch.tensor([[10, 30, 20], [60, 40, 50]]) >>> max_idx = torch.argmax(t) >>> torch.take_along_dim(t, max_idx) tensor([60]) >>> sorted_idx = torch.argsort(t, dim=1) >>> torch.take_along_dim(t, sorted_idx, dim=1) tensor([[10, 20, 30], [40, 50, 60]])
参数:
关键字参数:
相关用法
- Python PyTorch take用法及代码示例
- Python PyTorch tanh用法及代码示例
- Python PyTorch tan用法及代码示例
- Python PyTorch trunc用法及代码示例
- Python PyTorch tensorinv用法及代码示例
- Python PyTorch triu_indices用法及代码示例
- Python PyTorch tensor用法及代码示例
- Python PyTorch triangular_solve用法及代码示例
- Python PyTorch to_map_style_dataset用法及代码示例
- Python PyTorch trace_module用法及代码示例
- Python PyTorch topk用法及代码示例
- Python PyTorch tensorsolve用法及代码示例
- Python PyTorch tile用法及代码示例
- Python PyTorch transpose用法及代码示例
- Python PyTorch tensor_split用法及代码示例
- Python PyTorch t用法及代码示例
- Python PyTorch trapezoid用法及代码示例
- Python PyTorch tensordot用法及代码示例
- Python PyTorch tril_indices用法及代码示例
- Python PyTorch trace用法及代码示例
- Python PyTorch tril用法及代码示例
- Python PyTorch triu用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
- Python PyTorch cholesky用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.take_along_dim。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。