本文简要介绍python语言中 torch.linalg.tensorsolve
的用法。
用法:
torch.linalg.tensorsolve(A, B, dims=None, *, out=None) → Tensor
out(Tensor,可选的) -输出张量。如果
None
则忽略。默认值:None
。RuntimeError - 如果重新整形的
A
.view(m, m)
与上述m
不可逆,或者第一个ind
维度的乘积不等于其余维度的乘积。计算解决方案
X
到系统torch.tensordot(A, X) = B
。如果
m
是A
的第一个B
.ndim
维度的乘积,而n
是其余维度的乘积,则此函数期望m
和n
相等。返回的张量
x
满足tensordot(
A
, x, dims=x.ndim) ==
B
。x
具有形状A
[B.ndim:]
。如果指定
dims
,A
将被重新整形为A = movedim(A, dims, range(len(dims) - A.ndim + 1, 0))
支持 float、double、cfloat 和 cdouble dtypes 的输入。
例子:
>>> A = torch.eye(2 * 3 * 4).reshape((2 * 3, 4, 2, 3, 4)) >>> B = torch.randn(2 * 3, 4) >>> X = torch.linalg.tensorsolve(A, B) >>> X.shape torch.Size([2, 3, 4]) >>> torch.allclose(torch.tensordot(A, X, dims=X.ndim), B) True >>> A = torch.randn(6, 4, 4, 3, 2) >>> B = torch.randn(4, 3, 2) >>> X = torch.linalg.tensorsolve(A, B, dims=(0, 2)) >>> X.shape torch.Size([6, 4]) >>> A = A.permute(1, 3, 4, 0, 2) >>> A.shape[B.ndim:] torch.Size([6, 4]) >>> torch.allclose(torch.tensordot(A, X, dims=X.ndim), B, atol=1e-6) True
参数:
关键字参数:
抛出:
相关用法
- Python PyTorch tensorinv用法及代码示例
- Python PyTorch tensor用法及代码示例
- Python PyTorch tensor_split用法及代码示例
- Python PyTorch tensordot用法及代码示例
- Python PyTorch trunc用法及代码示例
- Python PyTorch triu_indices用法及代码示例
- Python PyTorch triangular_solve用法及代码示例
- Python PyTorch to_map_style_dataset用法及代码示例
- Python PyTorch trace_module用法及代码示例
- Python PyTorch topk用法及代码示例
- Python PyTorch tile用法及代码示例
- Python PyTorch tanh用法及代码示例
- Python PyTorch transpose用法及代码示例
- Python PyTorch take_along_dim用法及代码示例
- Python PyTorch t用法及代码示例
- Python PyTorch take用法及代码示例
- Python PyTorch trapezoid用法及代码示例
- Python PyTorch tan用法及代码示例
- Python PyTorch tril_indices用法及代码示例
- Python PyTorch trace用法及代码示例
- Python PyTorch tril用法及代码示例
- Python PyTorch triu用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
- Python PyTorch cholesky用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.linalg.tensorsolve。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。