Skip to content
Snippets Groups Projects
Select Git revision
  • 81dc62083cd3f2ec4fbe3b7751970ecb1e938a25
  • master default protected
  • fullUD
  • movementInAction
4 results

Oracle.cpp

Blame
  • Concat.cpp 283 B
    #include "Concat.hpp"
    
    ConcatImpl::ConcatImpl(int inputSize) : inputSize(inputSize)
    {
    }
    
    torch::Tensor ConcatImpl::forward(torch::Tensor input)
    {
      return input.view({input.size(0), -1});
    }
    
    int ConcatImpl::getOutputSize(int sequenceLength)
    {
      return sequenceLength * inputSize;
    }