对PyTorch torch.stack的实例讲解
来源:
中文源码网 浏览:308 次 日期:2024-03-29 09:39:05
【下载文档: 对PyTorch torch.stack的实例讲解.txt 】
对PyTorch torch.stack的实例讲解
不是concat的意思
import torch
a = torch.ones([1,2])
b = torch.ones([1,2])
torch.stack([a,b],1)
(0 ,.,.) =
1 1
1 1
[torch.FloatTensor of size 1x2x2]
以上这篇对PyTorch torch.stack的实例讲解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持中文源码网。