include_tensor_kernels.h

namespace ONNX_LIGHT_NAMESPACE

namespace onnx_backend_test

namespace kernel

class Concat

#include <include_tensor_kernels.h>

Concatenates a list of tensors along axis (since opset 13). All input tensors must share the same data type and the same shape except along the concatenation axis. axis may be negative, in which case it counts from the back of the input rank.

Public Functions

inline explicit Concat(const KernelContext &ctx)

Tensor operator()(const std::vector<Tensor> &inputs, int64_t axis) const

void operator()(const std::vector<Tensor> &inputs, int64_t axis, Tensor &output) const

Public Static Functions

static inline constexpr bool CanRunInPlace() noexcept

Private Members

KernelContext ctx_