1. torch.nn.functional官方解释
进入pytorch官方Docs
查看 torch.nn官方指南,讲的很详细
2 代码
1、
import torch
from torch import nn
input = torch.tensor([[1, 2, 0, 3, 1],
[0, 1, 2, 3, 1],
[1, 2, 1, 0, 0],
[5, 2, 3, 1, 1],
[