diff options
| -rw-r--r-- | tensorflow/go/op/wrappers.go | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go index 3d261c9d0a..09da8c1892 100644 --- a/tensorflow/go/op/wrappers.go +++ b/tensorflow/go/op/wrappers.go @@ -13882,6 +13882,14 @@ func DecodeCSVNaValue(value string) DecodeCSVAttr { } } +// DecodeCSVSelectCols sets the optional select_cols attribute to value. +// If not specified, defaults to <> +func DecodeCSVSelectCols(value []int64) DecodeCSVAttr { + return func(m optionalAttr) { + m["select_cols"] = value + } +} + // Convert CSV records to tensors. Each column maps to one tensor. // // RFC 4180 format is expected for the CSV records. |