Browse Source

fix name for array_ops.ones

tags/v0.12
Oceania2018 6 years ago
parent
commit
05b3172ddf
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      src/TensorFlowNET.Core/Operations/array_ops.py.cs

+ 1
- 2
src/TensorFlowNET.Core/Operations/array_ops.py.cs View File

@@ -260,8 +260,7 @@ namespace Tensorflow
return tf_with(ops.name_scope(name, "ones", new { dims }), scope =>
{
name = scope;
var shape = ops.convert_to_tensor(dims, dtype: TF_DataType.TF_INT32);
var output = gen_array_ops.fill(shape, constant_op.constant(1.0f, dtype: dtype), name: name);
var output = _constant_if_small(1, dims, dtype, name);
return output;
});
}


Loading…
Cancel
Save