PyTorch利用torch.cat()实现Tensor的拼接

Posted 2022-12-17 算法与编程之美

问题

方法

import torch
from torch import nn

conv1 = nn.Conv2d(
    in_channels=3,
    out_channels=32,
    kernel_size=3,
    stride=1,
    padding=1