tensorflow基操

分类: 文章 • 2025-04-20 23:14:52

tf.split(A,B,C)

A：表示切谁

B：表示切的方法，可以是list可以是常数

C：表示在那个维度进行切

tensorflow基操

注意TF卷积的输入格式[batch，长，宽，深]，red,gree,blue = tf.split(输入，3，3)第二个参数是常数3表示均匀切分，第三个通道是深度，表示第三个通道也就是对颜色通道均匀切分。

tf.concat(value,dim)

tensorflow基操