pytaco.tensor_add

pytaco.tensor_add(t1, t2, out_format, dtype=None)

Computes the element wise addition of two tensors.

  • If the two tensors are equal order, performs the operation element-wise

  • If the two tensors have order N and M and N > M, requires the last M dimensions of the tensor with order N be equal to the dimensions of the tensor with order M in order to broadcast.

The __add__ method in the tensor class is implemented using this method.

Parameters
t1, t2: tensors, array_like

tensors or array_like input operands.

out_format: format, mode_format
  • If a format is specified, the result tensor is stored in the format out_format.

  • If a mode_format is specified, the result the result tensor has a with all of the dimensions stored in the mode_format passed in.

dtype: Datatype, optional

The datatype of the output tensor.

Returns
sum: tensor

The element wise addition of the input tensors broadcasted as required.

Notes

The inner dimensions of the input tensor is broadcasted along the dimensions of whichever tensor has a higher order.