欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

torch常用函数(持续更新)

程序员文章站 2022-05-11 12:04:24
...

1. torch.cat()

	a = torch.ones([1, 2])
	b = torch.ones([1, 2])
	c = torch.cat([a, b], 1)

cat的第二个参数为1时,就是将a,b按列“拼在一起”,同于增广矩阵。
当参数为0时,将a,b按行“拼在一起”。