java icon indicating copy to clipboard operation
java copied to clipboard

How to convert DenseNDArray to NDArray ?

Open mullerhai opened this issue 3 years ago • 0 comments

HI , when I read TfRecordDataset from tfrecord file , I get the data type is DenseNDArray ,actual is DenseTString, I want to convert it to TString data.

my code -scala version

    val path = "/Volumes/Pink4T/transfer/code/github/stanford-tensorflow-tutorials/2017/data/friday.tfrecord"

    val dataset =  Dataset.tfRecordDataset(tf,path,"",0l).batch(10) // readData(path) //
//    DenseNdArray
    var count =0
    for (outputs <- dataset) {
      try {
        val batch1 = outputs.get(count).asTensor.asInstanceOf[TString]
        try {

          val byteNdArray:NdArray[Array[Byte]]= batch1.asBytes().asInstanceOf[NdArray[Array[Byte]]] //ByteNdArray]
         val bda = byteNdArray.asInstanceOf[DenseNdArray[TString]] // here !!! how to get   SparseNdArray?
          println(s"bda shape ${bda.shape()} ${kk.shape()}")

          count = count + 1
        } finally {
          if (batch1 != null) batch1.close()

        }
      }
    }

mullerhai avatar Jun 05 '22 13:06 mullerhai