log

nvtripy.log(input: Tensor) → Tensor

Computes the elementwise natural logarithm (base e) of the elements of the input tensor.

Parameters:

input (Tensor) – The input tensor.

Returns:

A new tensor of the same shape as the input tensor.

Return type:

Tensor

INPUT REQUIREMENTS:

input.dtype is one of [float32, float16, bfloat16]

OUTPUT GUARANTEES:

return[0].dtype == input.dtype

Example

input = tp.arange(1, 3, dtype=tp.float32)
output = tp.log(input)

Local Variables

>>> input
tensor([1, 2], dtype=float32, loc=gpu:0, shape=(2,))

>>> output
tensor([0, 0.693147], dtype=float32, loc=gpu:0, shape=(2,))