网站首页
编程语言
数据库
系统相关
其他分享
编程问答
24.5000
2024-09-22
torch.stack
看一下stack的直观解释,动词可以简单理解为:把……放成一堆、把……放成一摞。torch.stack方法用于沿着一个新的维度join(也可称为cat)一系列的张量(可以是2个张量或者是更多),它会插入一个新的维度,并让张量按照这个新的维度进行张量的cat操作。值得注意的是:张量序列中的张量必须要有相