本文简要介绍python语言中 torch.take
的用法。
用法:
torch.take(input, index) → Tensor
input(Tensor) -输入张量。
index(LongTensor) -张量的索引
在给定索引处返回具有
input
元素的新张量。输入张量被视为被视为一维张量。结果与索引的形状相同。例子:
>>> src = torch.tensor([[4, 3, 5], ... [6, 7, 8]]) >>> torch.take(src, torch.tensor([0, 2, 5])) tensor([ 4, 5, 8])
参数:
相关用法
- Python PyTorch take_along_dim用法及代码示例
- 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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。