Helloe veryone, I’ve been using tensorflow and numpy for a while, but now I wanted to dedicate myself to pytorch. My only problem is that I need to do a custom layer, which uses tensordot (https://www.tensorflow.org/api_docs/python/tf/tensordot) in tensorflow.
is there something in pytorch that I can use that is close to that tensordot?