单选题
在PyTorch中,如果要自定义一个神经网络层,需要继承()类。
A
torch.nn.Module
B
torch.Tensor
C
torch.optim.Optimizer
D
torch.autograFunction
答案解析
正确答案:A
解析:
暂无解析
相关知识点:
PyTorch自定义层继承类
题目纠错
人工智能训练师题库_1500题
