|
|
@@ -14,12 +14,9 @@ |
|
|
|
limitations under the License. |
|
|
|
******************************************************************************/ |
|
|
|
|
|
|
|
using System; |
|
|
|
using System.Collections.Generic; |
|
|
|
using System.Linq; |
|
|
|
using System.Text; |
|
|
|
using Tensorflow.Framework; |
|
|
|
using Tensorflow.Operations; |
|
|
|
using static Tensorflow.Python; |
|
|
|
|
|
|
|
namespace Tensorflow.Gradients |
|
|
@@ -99,7 +96,7 @@ namespace Tensorflow.Gradients |
|
|
|
new Tensor[] { non_neg_concat_dim, tf.constant(0) }, |
|
|
|
new Tensor[] { tf.constant(1), tf.constant(-1) }); |
|
|
|
var squeeze_sizes = array_ops.squeeze(slice); |
|
|
|
out_grads = gen_array_ops.split(grad, squeeze_sizes, non_neg_concat_dim).ToList(); |
|
|
|
out_grads = gen_array_ops.split(grad, squeeze_sizes, (int)non_neg_concat_dim.Data<int>()[0]).ToList(); |
|
|
|
} |
|
|
|
else |
|
|
|
{ |
|
|
|