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 array.

   let x = [1, 2, 3, 4] as [Float]
   let y = [5, 6, 7, 8] as [Float]

   let context = try BNNSGraph.makeContext {
       builder in

       let x = lhs.graphBuilderTensor(builder)

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

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