Unary

Unary operations for the Nabla framework.

nabla.ops.unary.negate(arg)

Element-wise negation.

nabla.ops.unary.cast(arg, dtype)

Cast array to different dtype.

nabla.ops.unary.sin(arg, dtype=None)

Element-wise sine.

nabla.ops.unary.cos(arg)

Element-wise cosine.

nabla.ops.unary.tanh(arg)

Element-wise hyperbolic tangent.

nabla.ops.unary.sigmoid(arg)

Element-wise sigmoid function.

nabla.ops.unary.abs(arg)

Element-wise absolute value.

nabla.ops.unary.floor(arg)

Element-wise floor function.

nabla.ops.unary.logical_not(arg)

Element-wise logical NOT operation.

nabla.ops.unary.incr_batch_dim_ctr(arg)

Increment batch dimension counter for debugging.

nabla.ops.unary.decr_batch_dim_ctr(arg)

Decrement batch dimension counter for debugging.

nabla.ops.unary.relu(arg)

Element-wise ReLU (Rectified Linear Unit) function.

nabla.ops.unary.log(arg)

Element-wise natural logarithm.

nabla.ops.unary.exp(arg)

Element-wise exponential function.

nabla.ops.unary.sqrt(arg)

Element-wise square root function.

Implemented as pow(arg, 0.5) for compatibility with the automatic differentiation system.

nabla.ops.unary.transfer_to(arg, device)

Transfer an array to a different device.