pytorch使用杂记

矩阵转换

Tensor.transpose(dim0, dim1) → Tensor

dim0, dim1维度分别转置然后返回,只能操作两维,且会直接修改底层数据

torch.permute(input, dims) → Tensor

翻转多维数组,返回视图。参数指定原张量的哪一维翻转到新维度上。

矩阵运算

torch.matmul(input, other, *, out=None) → Tensor

矩阵乘法

# pytorch 

评论

Your browser is out-of-date!

Update your browser to view this website correctly. Update my browser now

×