tensorflow 随笔记--把向量/vector转换为标量/scalar

numpy中把向量/vector转化为标量/scalar:

import numpy as np
vec = np.array([1]) #=> [1] 向量
scal = np.asscalar(vec)#=> 1 标量

tensorflow把向量转化为标量

import tensorflow as tf
vec = tf.Variable(tf.ones([1])) #=> [1]  向量
scal = tf.reshape(vec, []) # => 1   标量

猜你喜欢

转载自blog.csdn.net/qq_29007291/article/details/81178173