pytaco.tensor_div

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

Computes the element wise division 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 __truediv__ 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
quotient: tensor

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

Warning

This performs division on all elements in a tensor including the implicit zeros. Division by 0 produces undefined behavior.

Notes

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