本文简要介绍python语言中 torch.trace
的用法。
用法:
torch.trace(input) → Tensor
返回输入二维矩阵的对角线元素的总和。
例子:
>>> x = torch.arange(1., 10.).view(3, 3) >>> x tensor([[ 1., 2., 3.], [ 4., 5., 6.], [ 7., 8., 9.]]) >>> torch.trace(x) tensor(15.)
相关用法
- Python PyTorch trace_module用法及代码示例
- Python PyTorch transpose用法及代码示例
- Python PyTorch trapezoid用法及代码示例
- Python PyTorch trunc用法及代码示例
- Python PyTorch triu_indices用法及代码示例
- Python PyTorch triangular_solve用法及代码示例
- Python PyTorch tril_indices用法及代码示例
- Python PyTorch tril用法及代码示例
- Python PyTorch triu用法及代码示例
- Python PyTorch tensorinv用法及代码示例
- Python PyTorch tensor用法及代码示例
- Python PyTorch to_map_style_dataset用法及代码示例
- Python PyTorch topk用法及代码示例
- Python PyTorch tensorsolve用法及代码示例
- Python PyTorch tile用法及代码示例
- Python PyTorch tanh用法及代码示例
- Python PyTorch take_along_dim用法及代码示例
- Python PyTorch tensor_split用法及代码示例
- Python PyTorch t用法及代码示例
- Python PyTorch take用法及代码示例
- Python PyTorch tensordot用法及代码示例
- Python PyTorch tan用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
- Python PyTorch cholesky用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.trace。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。