This feature will allow us to build convnets. The idea is to use stride_trick and einsum.
It has to be determined whether we need a dedicated private for this operation, or we can rely on the utorch simplegrad.
Additional resources that may be useful:
This feature will allow us to build convnets. The idea is to use stride_trick and einsum.
It has to be determined whether we need a dedicated private for this operation, or we can rely on the utorch simplegrad.
Additional resources that may be useful: