torch.cos — PyTorch 2.7 documentation (original) (raw)

Shortcuts

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

Returns a new tensor with the cosine of the elements of input.

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

Parameters

input (Tensor) – the input tensor.

Keyword Arguments

out (Tensor, optional) – the output tensor.

Example:

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])