2021年11月8日
Pytorch中stack()方法的总结及理解
摘要： Torch.stack() 1. 概念 在一个新的维度上连接一个张量序列 2. 参数 tensors （sequence）需要连接的张量序列 dim （int）在第dim个维度上连接 注意输入的张量shape要完全一致，且dim必须小于len(tensors)。 3. 举例 3.1 四个shape为 阅读全文
posted @ 2021-11-08 22:27 MoSalah 阅读(2413) 评论(1) 推荐(0) 编辑