Class RealCooTensorDot

java.lang.Object

org.flag4j.linalg.ops.sparse.coo.real.RealCooTensorDot

public final class RealCooTensorDot
extends Object

Utility class for computing tensor dot products between two real sparse COO tensors.

Method Summary

Modifier and Type	Method	Description
static Tensor	tensorDot(CooTensor src1, CooTensor src2, int[] src1Axes, int[] src2Axes)	Computes the tensor contraction of this tensor with a specified tensor over the specified set of axes.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

tensorDot

public static Tensor tensorDot(CooTensor src1,
 CooTensor src2,
 int[] src1Axes,
 int[] src2Axes)

Computes the tensor contraction of this tensor with a specified tensor over the specified set of axes. That is, computes the sum of products between the two tensors along the specified set of axes.

Parameters:

src1 - First tensor in the contraction.

src2 - Second tensor in the contraction.

src1Axes - Axes along which to compute products for src1 tensor.

src2Axes - Axes along which to compute products for src2 tensor.

Returns:

The tensor dot product over the specified axes.

Throws:

IllegalArgumentException - If the two tensors shapes do not match along the specified axes pairwise in aAxes and bAxes.

IllegalArgumentException - If aAxes and bAxes do not match in length, or if any of the axes are out of bounds for the corresponding tensor.