Skip to content
Snippets Groups Projects
Select Git revision
  • d8fe2ce081e7c2236d78e46fd6ce42ba45357fa1
  • main default protected
2 results

class_ISTM.py

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;
    }