PyTorch函数

torch.clamp(input, min, max, out=None) → Tensor

将 input 张量中的每个元素值截断到区间 [min, max] 中。

torch.stack(tensors, dim=0, out=None) → Tensor

对 tensors 沿指定维度拼接,但会额外增加一维拼接的维度,即拼接时来自于 tensors 中每个 tensor 中的拼接元素单独组成一个维度。

torch.cat(tensors, dim=0, out=None) → Tensor

对 tensors 沿指定维度拼接,返回的 Tensor 维度不变,即拼接时直接将 tensors 中每个 tensor 的对应维度的元素拼接在一起。

torch.squeeze(input, dim=None, out=None) → Tensor

  • dim = None:去除 input 张量中所有 size 为 1 的维度。
  • 指定 dim 时:若该维度 size 为 1 则去除,否则保持 input 张量不变。

torch.unsqueeze(input, dim) → Tensor

  • dim >= 0:在指定维度前插入一维。
  • dim < 0:在指定维度后插入一维。

torch.flip(input, dims) → Tensor

  • 将 input 张量沿着列表/元组 dims 中的每一个维度依次翻转。

torch.Tensor.contiguous(memory_format=torch.contiguous_format) → Tensor

  • 返回一个内存连续且有相同数据的 Tensor,如果原 Tensor 已经内存连续,则直接返回原 Tensor。

torch.Tensor.permute(*dims) → Tensor

  • 根据 dims 给定的维度顺序对张量进行维度换位。

torch.Tensor.transpose(dim0, dim1) → Tensor

  • 对 dim0 和 dim1 两个维度换位。

torch.nn.functional.grid_sample(input, grid, mode='bilinear', padding_mode='zeros', align_corners=None) → Tensor


本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!