concat.h

Go to the documentation of this file.

#pragma once

#include "tensor/tensor.h"

namespace magmadnn {
namespace math {

template <typename T>
void concat(Tensor<T> *A, Tensor<T> *B, Tensor<T> *C, unsigned int axis);

}
}