minsky::anonymous_namespace{minskyTensorOps.cc} Namespace Reference

Classes
struct	is_equal
struct	TensorOpError

Functions
void	meldArgsIntoCommonHypercube (vector< TensorPtr > &args)

Function Documentation

meldArgsIntoCommonHypercube()

void minsky::anonymous_namespace{minskyTensorOps.cc}::meldArgsIntoCommonHypercube

(

vector< TensorPtr > &

args

)

Definition at line 1322 of file minskyTensorOps.cc.

Referenced by minsky::GeneralTensorOp< OperationType::meld >::setArguments(), and minsky::GeneralTensorOp< OperationType::merge >::setArguments().

                                                              {
      Hypercube hc;
      for (auto& i: args)
        unionHypercube(hc, i->hypercube(), false);

      // list of final dimension names in order
      vector<string> unionDims; unionDims.reserve(hc.xvectors.size());
      for (auto& i: hc.xvectors) unionDims.push_back(i.name);

      // create tensor chains for each argument to permute it into the common hypercube
      for (auto& i: args)
        {
          set <string> argDims;
          for (auto& j: i->hypercube().xvectors) argDims.insert(j.name);
          auto spread=make_shared<SpreadLast>();
          Hypercube spreadHC;
          for (auto& j: hc.xvectors)
            if (!argDims.contains(j.name))
              spreadHC.xvectors.push_back(j);
          spread->setArgument(i,{});
          spread->setSpreadDimensions(spreadHC);

          auto pivot=make_shared<Pivot>();
          pivot->setArgument(spread,{});
          pivot->setOrientation(unionDims);

          if (pivot->hypercube()==hc)
            i=pivot;
          else
            {
              auto spreadOverHC=make_shared<SpreadOverHC>();
              spreadOverHC->hypercube(hc);
              spreadOverHC->setArgument(pivot,{});
              i=spreadOverHC;
            }
        }
    }

Here is the caller graph for this function: