Skip to content

Math Functions

simplegrad.functions.math.log(x: Tensor) -> Tensor

Compute element-wise natural logarithm.

Parameters:

Name Type Description Default
x Tensor

Input tensor. All values must be positive.

 required

Returns:

Type Description
Tensor

Tensor of ln(x).

Raises:

Type Description
ValueError

If any value in x is <= 0 (checked in eager mode only).
Source code in simplegrad/functions/math.py 
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
def log(x: Tensor) -> Tensor:
    """Compute element-wise natural logarithm.

    Args:
        x: Input tensor. All values must be positive.

    Returns:
        Tensor of ln(x).

    Raises:
        ValueError: If any value in x is <= 0 (checked in eager mode only).
    """
    if not is_lazy() and np.any(x.values <= 0):
        raise ValueError("Log of negative value is undefined")
    return _Log.apply(x)

simplegrad.functions.math.exp(x: Tensor) -> Tensor

Compute element-wise exponential.

Parameters:

Name Type Description Default
x Tensor

Input tensor.

 required

Returns:

Type Description
Tensor

Tensor of e^x.
Source code in simplegrad/functions/math.py 
89
90
91
92
93
94
95
96
97
98
def exp(x: Tensor) -> Tensor:
    """Compute element-wise exponential.

    Args:
        x: Input tensor.

    Returns:
        Tensor of e^x.
    """
    return _Exp.apply(x)

simplegrad.functions.math.sin(x: Tensor) -> Tensor

Compute element-wise sine.

Parameters:

Name Type Description Default
x Tensor

Input tensor (radians).

 required

Returns:

Type Description
Tensor

Tensor of sin(x).
Source code in simplegrad/functions/math.py 
101
102
103
104
105
106
107
108
109
110
def sin(x: Tensor) -> Tensor:
    """Compute element-wise sine.

    Args:
        x: Input tensor (radians).

    Returns:
        Tensor of sin(x).
    """
    return _Sin.apply(x)

simplegrad.functions.math.cos(x: Tensor) -> Tensor

Compute element-wise cosine.

Parameters:

Name Type Description Default
x Tensor

Input tensor (radians).

 required

Returns:

Type Description
Tensor

Tensor of cos(x).
Source code in simplegrad/functions/math.py 
113
114
115
116
117
118
119
120
121
122
def cos(x: Tensor) -> Tensor:
    """Compute element-wise cosine.

    Args:
        x: Input tensor (radians).

    Returns:
        Tensor of cos(x).
    """
    return _Cos.apply(x)

simplegrad.functions.math.tan(x: Tensor) -> Tensor

Compute element-wise tangent.

Parameters:

Name Type Description Default
x Tensor

Input tensor (radians).

 required

Returns:

Type Description
Tensor

Tensor of tan(x).
Source code in simplegrad/functions/math.py 
125
126
127
128
129
130
131
132
133
134
def tan(x: Tensor) -> Tensor:
    """Compute element-wise tangent.

    Args:
        x: Input tensor (radians).

    Returns:
        Tensor of tan(x).
    """
    return _Tan.apply(x)