按照行或者列分割一个矩阵
第一个参数是待切分的张量，第二个是切分的大小，共切分为三个，每一个分别为4列，15列，11列。第三个参数为1表示按照列切分，为0表示按照行切分。
切分的子块的大小加起来应该等于母块的大小。比如这个例子，4+15+11=30列
看下面的代码例子
tf.split用法
猜你喜欢
转载自blog.csdn.net/qq_31150463/article/details/84137883
今日推荐
周排行