矩阵转换
Tensor.transpose(dim0, dim1) → Tensor
对dim0
, dim1
维度分别转置然后返回,只能操作两维,且会直接修改底层数据
torch.permute(input, dims) → Tensor
翻转多维数组,返回视图。参数指定原张量的哪一维翻转到新维度上。
矩阵运算
torch.matmul(input, other, *, out=None) → Tensor
矩阵乘法
Tensor.transpose(dim0, dim1) → Tensor
对dim0
, dim1
维度分别转置然后返回,只能操作两维,且会直接修改底层数据
torch.permute(input, dims) → Tensor
翻转多维数组,返回视图。参数指定原张量的哪一维翻转到新维度上。
torch.matmul(input, other, *, out=None) → Tensor
矩阵乘法