pytorch之torch.nn.functional.grid_sample 函数详解

目录

    • 参考

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

用于执行网格采样。给定一个输入张量和一个表示流场的网格,该函数使用输入值和来自网格的像素位置计算输出。

目前,仅支持空间(4 维)和体积(5 维)输入。

  • 空间 (4D) 输入
    对于形状为 ( N , C , H i n , W i n ) (N, C, H_{in}, W_{in}) (N,C,Hin,Win) 的输入和形状为 ( N , H o u t , W o u t , 2 ) (N, H_{out}, W_{out}, 2) (N,Hout,W

你可能感兴趣的:(pytorch,pytorch,深度学习,人工智能)