RealCooTensorDot

java.lang.Object

org.flag4j.operations.sparse.coo.real.RealCooTensorDot

public final class RealCooTensorDot extends Object

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

Constructor Summary

Constructors

Modifier

Constructor

Description

private

RealCooTensorDot()
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

Constructor Details
- RealCooTensorDot
  
  private RealCooTensorDot()
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.