torch常用函数(持续更新)

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按行“拼在一起”。


版权声明:本文为qq_44721124原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。