本文简要介绍python语言中 torch.index_select
的用法。
用法:
torch.index_select(input, dim, index, *, out=None) → Tensor
out(Tensor,可选的) -输出张量。
返回一个新张量,该张量使用
index
中的条目沿维度dim
索引input
张量,该条目是LongTensor
。返回的张量与原始张量 (
input
) 具有相同的维数。第dim
维度的大小与index
的长度相同;其他维度的大小与原始张量中的大小相同。注意
返回的张量不是使用与原始张量相同的存储。如果
out
如果形状与预期不同,我们会默默地将其更改为正确的形状,并在必要时重新分配底层存储。例子:
>>> x = torch.randn(3, 4) >>> x tensor([[ 0.1427, 0.0231, -0.5414, -1.0009], [-0.4664, 0.2647, -0.1228, -1.1068], [-1.1734, -0.6571, 0.7230, -0.6004]]) >>> indices = torch.tensor([0, 2]) >>> torch.index_select(x, 0, indices) tensor([[ 0.1427, 0.0231, -0.5414, -1.0009], [-1.1734, -0.6571, 0.7230, -0.6004]]) >>> torch.index_select(x, 1, indices) tensor([[ 0.1427, -0.5414], [-0.4664, -0.1228], [-1.1734, 0.7230]])
参数:
关键字参数:
相关用法
- Python PyTorch invoke_on_rank_and_broadcast_result用法及代码示例
- Python PyTorch inv_ex用法及代码示例
- Python PyTorch inv用法及代码示例
- Python PyTorch inner用法及代码示例
- Python PyTorch inference_mode用法及代码示例
- Python PyTorch ignore用法及代码示例
- Python PyTorch ihfft用法及代码示例
- Python PyTorch identity用法及代码示例
- Python PyTorch import_huggingface_model用法及代码示例
- Python PyTorch is_tensor_like用法及代码示例
- Python PyTorch i0用法及代码示例
- Python PyTorch irfft用法及代码示例
- Python PyTorch is_nonzero用法及代码示例
- Python PyTorch isneginf用法及代码示例
- Python PyTorch ifft2用法及代码示例
- Python PyTorch ifftn用法及代码示例
- Python PyTorch i1e用法及代码示例
- Python PyTorch is_scripting用法及代码示例
- Python PyTorch isclose用法及代码示例
- Python PyTorch ifftshift用法及代码示例
- Python PyTorch isnan用法及代码示例
- Python PyTorch imag用法及代码示例
- Python PyTorch is_tensor_method_or_property用法及代码示例
- Python PyTorch isreal用法及代码示例
- Python PyTorch import_fairseq_model用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.index_select。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。