tensorflow常用函数

import tensorflow as tf
import numpy as np

tf.placeholder()
tf.variable()

矩阵乘法

tf.matual()

数字乘法

tf.mutiply()

tf.reduce_sum()

reduce_sum(
    input_tensor,
    axis=None,
    keep_dims=False,
    name=None,
    reduction_indices=None
)

input_tensor:表示输入
axis:表示在那个维度进行sum操作。axis=0纵轴, axis=1横轴
keep_dims:表示是否保留原始数据的维度,False相当于执行完后原始数据就会少一个维度。
reduction_indices:为了跟旧版本的兼容,现在已经不使用了。

tf.reduce_mean()
同上

猜你喜欢

转载自blog.csdn.net/baidu_27643275/article/details/79670308