torch.foreach.sub

torch.foreach.sub(inputs: TensorList, other: Scalar, /) → tuple[Tensor, ...]

torch.foreach.sub(inputs: TensorList, other: ScalarList, /) → tuple[Tensor, ...]

torch.foreach.sub(inputs: TensorList, other: TensorList, /, *, alpha: Scalar = 1) → tuple[Tensor, ...]

Applies torch.sub() to every tensor in inputs.

This is semantically equivalent to applying torch.sub() independently at every list position. Does not mutate its arguments and returns a tuple of result tensors.

Tensor-list arguments must be non-empty. Corresponding tensor or scalar lists must have the same length. An accelerated multi-tensor implementation is used only when supported by the inputs; otherwise the operation falls back to per-tensor execution.

Parameters:

• inputs (list or tuple of Tensor) – tensors to transform.

• other (Number, list or tuple of Number, or list or tuple of Tensor) – operand shared across positions or supplied per position.

• alpha (Number, optional) – supported only when other is a tensor list. Default: 1.

Returns:

a tuple containing one result tensor for each input tensor.