graphBuilderTensor(_:)

Declaration

func graphBuilderTensor(_ builder: BNNSGraph.Builder) -> BNNSGraph.Builder.Tensor<Element>

Discussion

The following code shows how to use this function to register a tensor from an unsafe buffer pointe:.

   let context = try BNNSGraph.makeContext {
       builder in

       ([1, 2, 3, 4] as [Float]).withUnsafeBufferPointer { x in
           ([5, 6, 7, 8] as [Float]).withUnsafeBufferPointer { y in


               let x = x.graphBuilderTensor(builder)

               let z = x.matmul(transpose: true,
                                other: y)

               return [z] // On return, `z[0]` equals `70`.
           }
       }
   }