分类：后端backend

2020
10-08

浅谈keras中的后端backend及其相关函数(K.prod,K.cast)

一、K.prodprodkeras.backend.prod(x,axis=None,keepdims=False)功能：在某一指定轴，计算张量中的值的乘积。参数x:张量或变量。axis:一个整数需要计算乘积的轴。keepdims:布尔值，是否保留原尺寸。如果keepdims为False，则张量的秩减1。如果keepdims为True，缩小的维度保留为长度1。返回x的元素的乘积的张量。Numpy实现defprod(x,axis=None,keepdims=False):ifisinstance(axis,list):axis=tuple(ax... 继续阅读 >