TensorFlow Swift simulation

TensorFlowSimulation.swift

import Foundation
import TensorFlow
import numsw

var tf = TensorFlow()

let a = tf.constant(NDArray<Float>(shape: [3], elements: [5, 7, 10]))
let b = tf.constant(NDArray<Float>(shape: [3], elements: [2, 3, 21]))
let c = tf.constant(NDArray<Float>(shape: [3], elements: [3, 5, 7]))

var d = a * b  // tf.multiply(a,b)
var e = c + b  // tf.add(c,b)
var f = d - e  // tf.subtract(d,e)

tf.run(f)