fn

sub

Tensor
sub(input: Tensor, other: Tensor | Scalar)
source

Element-wise subtraction with broadcasting and dtype promotion.

Computes input - other element-wise. Broadcasting and dtype promotion follow the standard rules.

Parameters

inputTensor
Left operand.
otherTensor or scalar
Right operand. Broadcasts against input following the standard broadcasting rules; Python scalars are promoted to a tensor of matching dtype.

Returns

Tensor

Element-wise result with shape broadcast(input.shape, other.shape) and dtype determined by the usual type-promotion rules.

Notes

Mathematical definition:

outi=inputiotheri\text{out}_i = \text{input}_i - \text{other}_i

Autograd: out/input=1\partial \text{out}/\partial \text{input} = 1, out/other=1\partial \text{out}/\partial \text{other} = -1.

Examples

>>> import lucid
>>> a = lucid.tensor([1.0, 2.0, 3.0])
>>> b = lucid.tensor([4.0, 5.0, 6.0])
>>> lucid.sub(a, b)
Tensor([...])