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

Posted 算法与编程之美

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了PyTorch利用torch.cat()实现Tensor的拼接相关的知识，希望对你有一定的参考价值。

问题

方法


import torch
from torch import nn

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

以上是关于PyTorch利用torch.cat()实现Tensor的拼接的主要内容，如果未能解决你的问题，请参考以下文章