Pytorch中的view()与view_as()

本人是pytorch 小白, 对函数的理解都是自己瞎琢磨的, 如果有不对的地方, 希望大神指正~

1 view(*args)函数用于变换 tensor 矩阵的类型

里面有一个矩阵, 是两行三列, 因此 torchsize 是 [1, 2, 3]

Pytorch中的view()与view_as()_第1张图片

将 a 变换为三个矩阵, 两行一列, 因此torchsize 是 [3, 2, 1]

Pytorch中的view()与view_as()_第2张图片

2 view_as(*arg) 函数用于依据 *arg 来改变矩阵形状, 其中 *arg 是一个矩阵

Pytorch中的view()与view_as()_第3张图片

你可能感兴趣的:(pytorch)