Skip to content

Commit 41abe69

Browse files
Update tensor.hh
1 parent d7a9753 commit 41abe69

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

src/tensor-array/core/tensor.hh

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,12 @@ namespace tensor_array
9090
dilation;
9191
};
9292

93+
class Tensor;
94+
95+
TENSOR_ARRAY_API Tensor tensor_rand(const std::initializer_list<unsigned int>&, unsigned int = std::rand());
96+
97+
TENSOR_ARRAY_API std::pair<Tensor, Tensor> tensor_broadcasting(const Tensor&, const Tensor&, unsigned char 0, unsigned char = 0);
98+
9399
/**
94100
* \brief Dynamic derivative tensor.
95101
* \brief This class use to calculate the tensor.
@@ -256,9 +262,9 @@ namespace tensor_array
256262

257263
friend TENSOR_ARRAY_API Tensor add_dim(const std::vector<Tensor>&);
258264

259-
friend TENSOR_ARRAY_API Tensor tensor_rand(const std::initializer_list<unsigned int>&, unsigned int/* = std::rand() */);
265+
friend Tensor tensor_rand(const std::initializer_list<unsigned int>&, unsigned int/* = std::rand() */);
260266

261-
friend TENSOR_ARRAY_API std::pair<Tensor, Tensor> tensor_broadcasting(const Tensor&, const Tensor&, unsigned char /*= 0*/, unsigned char /*= 0*/);
267+
friend std::pair<Tensor, Tensor> tensor_broadcasting(const Tensor&, const Tensor&, unsigned char /*= 0*/, unsigned char /*= 0*/);
262268

263269
friend TENSOR_ARRAY_API std::ostream& operator<<(std::ostream&, const Tensor&);
264270

0 commit comments

Comments
 (0)