new_empty_tensor_op.h 212 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
#pragma once

#include <ATen/ATen.h>

namespace vision {
namespace ops {

at::Tensor new_empty_tensor(
    const at::Tensor& input,
    const c10::List<int64_t>& shape);

} // namespace ops
} // namespace vision