Python PyTorch trace用法及代碼示例

本文簡要介紹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.)