tensor_gen.cpp File Reference

#include <vector>
#include <iostream>
#include <fstream>
#include <stdexcept>
#include <functional>
#include <algorithm>
#include <H5Cpp.h>

Go to the source code of this file.

Data Structures
class	Tensor

Functions
int	main (int argc, char *argv[])

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 167 of file tensor_gen.cpp.

{
  if (argc < 4)
  {
    cout << "Usage: " << argv[0] << " <tensor name> <output file name> dim0, dim1, dim2, ..."
         << endl;
    cout << "Where: dim0, dim1, dim2, ... are the generated tensor shape dimensions" << endl;
    return 1;
  }
 
  vector<hsize_t> dimensions;
 
  for (int i = 3; i < argc; ++i)
  {
    try
    {
      int d = stoi(argv[i]);
 
      if (d <= 0)
      {
        cout << "The dimensions must be positive values. This is not a correct dimension value: "
             << d << endl;
        return 1;
      }
 
      dimensions.push_back(d);
    }
    catch (const invalid_argument &)
    {
      cout << "The parameter does not look as an integer value: " << argv[i] << endl;
      return 1;
    }
    catch (const out_of_range &)
    {
      cout << "The value is out of the C++ \"int\" type range: " << argv[i] << endl;
      return 1;
    }
  }
 
  Tensor caffe_tensor(dimensions);
  fillTensor(caffe_tensor);
  writeTensorToHDF5File(dimensions, argv[1], "in_" + string(argv[2]) + "_caffe", caffe_tensor);
 
  vector<hsize_t> tf_reshape{0, 2, 3, 1};
  Tensor tf_tensor = caffe_tensor.transpose(tf_reshape);
  writeTensorToDatFile(string(argv[2]) + "_tf", tf_tensor);
 
  return 0;
}

References Tensor::transpose().