Confused with conv2d op
Hi everyone,
I'm trying to test out the conv2d mojo op in the graph API: https://docs.modular.com/max/api/mojo/graph/ops/convolution/conv2d/ but I'm pretty confused and there are some things I've been trying to figure out for a while now.
First of all I don't know what is the RSCF layout and I can't seem to find any information online. This leads me to not really understand how to know which shape the output tensor will have given the shape of and input and a filter. I'm currently running this code ( I will append it in a comment because the message is to large)
The code works, but if I change the alias on top for other numbers that in my mind make sense I start getting errors. I've also been looking for example everywhere but I haven't been able to find any.
I don't know if I'm making some rookie mistake or If I'm missing some knowledge about convolution so any help is extremely appreciated!. .
I'm trying to test out the conv2d mojo op in the graph API: https://docs.modular.com/max/api/mojo/graph/ops/convolution/conv2d/ but I'm pretty confused and there are some things I've been trying to figure out for a while now.
First of all I don't know what is the RSCF layout and I can't seem to find any information online. This leads me to not really understand how to know which shape the output tensor will have given the shape of and input and a filter. I'm currently running this code ( I will append it in a comment because the message is to large)
The code works, but if I change the alias on top for other numbers that in my mind make sense I start getting errors. I've also been looking for example everywhere but I haven't been able to find any.
I don't know if I'm making some rookie mistake or If I'm missing some knowledge about convolution so any help is extremely appreciated!. .
