Template Function at::symint::conv2d(const at::Tensor&, const at::Tensor&, const ::std::optional<at::Tensor>&, c10::SymIntArrayRef, c10::SymIntArrayRef, c10::SymIntArrayRef, c10::SymInt)¶
Defined in File Functions.h
Function Documentation¶
-
template<typename T, typename = std::enable_if_t<std::is_same_v<T, c10::SymInt>>>
at::Tensor at::symint::conv2d(const at::Tensor &input, const at::Tensor &weight, const ::std::optional<at::Tensor> &bias = {}, c10::SymIntArrayRef stride = c10::SymInt(1), c10::SymIntArrayRef padding = c10::SymInt(0), c10::SymIntArrayRef dilation = c10::SymInt(1), c10::SymInt groups = 1)¶