【torch.nn.functional.conv3d】【算子问题】
收藏回复举报
【torch.nn.functional.conv3d】【算子问题】
t('forum.solved') 已解决
发表于2025-04-25 17:48:50
0 查看

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)

报错截图:

cke_3381.png

我要发帖子