tensorflow常用numpy.ndarray, list操

2020-03-23  本文已影响0人  啊啊啊啊啊1231

tf.reduce_mean 函数用于计算张量tensor沿着指定的数轴(tensor的某一维度)上的的平均值,主要用作降维或者计算tensor(图像)的平均值。

reduce_mean(input_tensor,

                axis=None,

                keep_dims=False,

                name=None,

                reduction_indices=None)

tf.reduce_sum :计算tensor指定轴方向上的所有元素的累加和;

tf.reduce_max  :  计算tensor指定轴方向上的各个元素的最大值;

tf.reduce_all :  计算tensor指定轴方向上的各个元素的逻辑和(and运算);

tf.reduce_any:  计算tensor指定轴方向上的各个元素的逻辑或(or运算);

tensorflow推荐模型中定义模型初始化,通常另外写一个.py文件

class KGCN(object):

self.user_indices = tf.placeholder(dtype=tf.int64, shape=[None], name='user_indices')

self.item_indices = tf.placeholder(dtype=tf.int64, shape=[None], name='item_indices')

self.labels = tf.placeholder(dtype=tf.float32, shape=[None], name='labels')

上一篇 下一篇

猜你喜欢

热点阅读