def conv2d_custom

stelliferous@0.0.2.0 · tensor_f32.bend:282

import Base
import stelliferous@0.0.2.0/math_activation.bend as Activations
import stelliferous@0.0.2.0/tensor_convolution.bend as TensorConvolution

282# conv2d with a custom activation of the given scalar operations per element.
283def conv2d_custom(~activation: F32 -> F32,operations: F32,input: Tensor(),weights: Tensor(),bias: Tensor(),stride: U32,padding: U32) -> Tensor():
284  TensorConvolution.conv2d(~activation,Activations.Custom{operations},input,weights,bias,stride,padding)