The tensordot function#

Aliases#

halerium.core.tensordot
halerium.core.math.tensordot
tensordot(left, right, axes=1)#

Tensor dot product along specified axes.

Parameters:
  • left – The left operand of the tensor dot operation.

  • right – The right operand of the tensor dot operation.

  • axes (int, list or tuple of lists or tuples of ints) – The axes to sum over. For an int N, sum over last N axes of left and first N axes of right. Otherwise, axes[0] specifies the axes of left and axes[1] specifies the axes of right to sum over.

Returns:

operator – The halerium Operator representing the operation.

Return type:

halerium.core.operator.TensorDot