No handlers found: "aten::einsum" about torchprofile HOT 4 OPEN

The model in my code is a bit complex:

A = torch.randn(b,n,c,h,w,k)
B = torch.randn(b,n,c,h,w,k)
C = torch.einsum('bnchwk,bnchwk -> bnchw', A, B)
C.shape
torch.Size([b,n,c,h,w])

A simple example:

import torch
A = torch.randn(3, 4)
B = torch.randn(4, 5)
C = torch.einsum('ik,kj->ij', A, B)
C.shape
torch.Size([3, 5])

Thank you!

from torchprofile.

Could you please provide an example model that contains aten::permute? Thank you!

from torchprofile.

Thank you for your reply!

x = torch.randn(2, 3, 5)
x.size()
torch.Size([2, 3, 5])
x.permute(2, 0, 1).size()
torch.Size([5, 2, 3])

It seems to not require any multiplication and addition operations.
I have calculated flops using other calculation codes, thank you!

from torchprofile.

Thanks! I was trying to refer to aten::einsum. Do you have an example of that?

from torchprofile.