Unary Operations#

Element-wise unary operations that operate on a single array.

Quick Reference#

abs#

nabla.abs(arg: nabla.core.array.Array) -> nabla.core.array.Array

Nabla operation: abs

cast#

nabla.cast(arg: nabla.core.array.Array, dtype: max._core.dtype.DType) -> nabla.core.array.Array

Cast array elements to a different data type.

cos#

nabla.cos(arg: nabla.core.array.Array) -> nabla.core.array.Array

Element-wise cosine function.

decr_batch_dim_ctr#

nabla.decr_batch_dim_ctr(arg: nabla.core.array.Array) -> nabla.core.array.Array

Nabla operation: decr_batch_dim_ctr

exp#

nabla.exp(arg: nabla.core.array.Array) -> nabla.core.array.Array

Element-wise exponential function.

floor#

nabla.floor(arg: nabla.core.array.Array) -> nabla.core.array.Array

Nabla operation: floor

incr_batch_dim_ctr#

nabla.incr_batch_dim_ctr(arg: nabla.core.array.Array) -> nabla.core.array.Array

Nabla operation: incr_batch_dim_ctr

log#

nabla.log(arg: nabla.core.array.Array) -> nabla.core.array.Array

Element-wise natural logarithm.

logical_not#

nabla.logical_not(arg: nabla.core.array.Array) -> nabla.core.array.Array

Nabla operation: logical_not

negate#

nabla.negate(arg: nabla.core.array.Array) -> nabla.core.array.Array

Element-wise negation.

relu#

nabla.relu(arg: nabla.core.array.Array) -> nabla.core.array.Array

Element-wise rectified linear unit activation.

sigmoid#

nabla.sigmoid(arg: nabla.core.array.Array) -> nabla.core.array.Array

Nabla operation: sigmoid

sin#

nabla.sin(arg: nabla.core.array.Array, dtype: max._core.dtype.DType | None = None) -> nabla.core.array.Array

Element-wise sine function.

sqrt#

nabla.sqrt(arg: nabla.core.array.Array) -> nabla.core.array.Array

Nabla operation: sqrt

tanh#

nabla.tanh(arg: nabla.core.array.Array) -> nabla.core.array.Array

Nabla operation: tanh

transfer_to#

nabla.transfer_to(arg: nabla.core.array.Array, device: max._core.driver.Device) -> nabla.core.array.Array

Nabla operation: transfer_to