Python PyTorch cos用法及代碼示例

本文簡要介紹python語言中 torch.cos 的用法。

用法:

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

參數：

input(Tensor) -輸入張量。

關鍵字參數：

out(Tensor,可選的) -輸出張量。

返回具有 input 元素餘弦的新張量。

\text{out}_{i} = \cos(\text{input}_{i})

例子：

>>> a = torch.randn(4)
>>> a
tensor([ 1.4309,  1.2706, -0.8562,  0.9796])
>>> torch.cos(a)
tensor([ 0.1395,  0.2957,  0.6553,  0.5574])