cann版本:8.0.0
torch_npu:2.1.0.post10
soc_version:910b3
代码:
import torch
import torch_npu
import torch.nn.functional as F
filters = torch.randn(33, 16, 3, 3, 3).to("npu:0")
inputs = torch.randn(20, 16, 50, 10, 20).to("npu:0")
F.conv3d(inputs, filters)
报错截图:

cann版本:8.0.0
torch_npu:2.1.0.post10
soc_version:910b3
代码:
import torch
import torch_npu
import torch.nn.functional as F
filters = torch.randn(33, 16, 3, 3, 3).to("npu:0")
inputs = torch.randn(20, 16, 50, 10, 20).to("npu:0")
F.conv3d(inputs, filters)
报错截图: