PyTorch 中的 take()、take_along_dim() 和 Gather()

take()可以使用 0D 或多个 D 张量从 0D 或多个 D 张量中获取一个或多个元素，如下所示：

*备注：

take()可以从torch和张量中调用。
第二个参数（必需）是 0D 或更多 D 张量torch。
第一个参数（必需）是具有张量的 0D 或更多 D 张量。
第二个参数的张量torch或第一个参数的张量决定返回张量的大小。

import torch

my_tensor = torch.tensor([[9, 5, 0, 6, 2