tensorflow
Matrix och vektor aritmetik
Sök…
Elementvis multiplikation
För att utföra elementvis multiplikation på tensorer kan du använda något av följande:
-
a*b
-
tf.multiply(a, b)
Här är ett fullständigt exempel på elementvis multiplikation med båda metoderna.
import tensorflow as tf import numpy as np # Build a graph graph = tf.Graph() with graph.as_default(): # A 2x3 matrix a = tf.constant(np.array([[ 1, 2, 3], [10,20,30]]), dtype=tf.float32) # Another 2x3 matrix b = tf.constant(np.array([[2, 2, 2], [3, 3, 3]]), dtype=tf.float32) # Elementwise multiplication c = a * b d = tf.multiply(a, b) # Run a Session with tf.Session(graph=graph) as session: (output_c, output_d) = session.run([c, d]) print("output_c") print(output_c) print("\noutput_d") print(output_d)
Skriver ut följande:
output_c
[[ 2. 4. 6.]
[ 30. 60. 90.]]
output_d
[[ 2. 4. 6.]
[ 30. 60. 90.]]
Scalar Times a Tensor
I följande exempel multipliceras en 2 med 3-tensor med ett skalvärde (2).
# Build a graph
graph = tf.Graph()
with graph.as_default():
# A 2x3 matrix
a = tf.constant(np.array([[ 1, 2, 3],
[10,20,30]]),
dtype=tf.float32)
# Scalar times Matrix
c = 2 * a
# Run a Session
with tf.Session(graph=graph) as session:
output = session.run(c)
print(output)
Detta skriver ut
[[ 2. 4. 6.]
[ 20. 40. 60.]]
Punkt produkt
Punktprodukten mellan två tensorer kan utföras med hjälp av:
tf.matmul(a, b)
Ett fullständigt exempel ges nedan:
# Build a graph
graph = tf.Graph()
with graph.as_default():
# A 2x3 matrix
a = tf.constant(np.array([[1, 2, 3],
[2, 4, 6]]),
dtype=tf.float32)
# A 3x2 matrix
b = tf.constant(np.array([[1, 10],
[2, 20],
[3, 30]]),
dtype=tf.float32)
# Perform dot product
c = tf.matmul(a, b)
# Run a Session
with tf.Session(graph=graph) as session:
output = session.run(c)
print(output)
skriver ut
[[ 14. 140.]
[ 28. 280.]]
Modified text is an extract of the original Stack Overflow Documentation
Licensierat under CC BY-SA 3.0
Inte anslutet till Stack Overflow