org
.
platanios
.
tensorflow
.
api
.
tensors
.
ops
NN
Related Doc:
package ops
object
NN
extends
NN
Linear Supertypes
NN
,
AnyRef
,
Any
Ordering
Alphabetic
By Inheritance
Inherited
NN
NN
AnyRef
Any
Hide All
Show All
Visibility
Public
All
Type Members
case class
NNOps
(
tensor:
Tensor
)
extends
Product
with
Serializable
Value Members
final
def
!=
(
arg0:
Any
)
:
Boolean
Definition Classes
AnyRef → Any
final
def
##
()
:
Int
Definition Classes
AnyRef → Any
final
def
==
(
arg0:
Any
)
:
Boolean
Definition Classes
AnyRef → Any
def
addBias
(
value:
Tensor
,
bias:
Tensor
,
cNNDataFormat:
CNNDataFormat
=
CNNDataFormat.default
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
final
def
asInstanceOf
[
T0
]
:
T0
Definition Classes
Any
def
clone
()
:
AnyRef
Attributes
protected[
java.lang
]
Definition Classes
AnyRef
Annotations
@throws
(
...
)
def
conv2D
(
input:
Tensor
,
filter:
Tensor
,
stride1:
Long
,
stride2:
Long
,
padding:
PaddingMode
,
dataFormat:
CNNDataFormat
=
CNNDataFormat.default
,
useCuDNNOnGPU:
Boolean
=
true
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
conv2DBackpropFilter
(
input:
Tensor
,
filterSizes:
Tensor
,
outputGradient:
Tensor
,
stride1:
Long
,
stride2:
Long
,
padding:
PaddingMode
,
dataFormat:
CNNDataFormat
=
CNNDataFormat.default
,
useCuDNNOnGPU:
Boolean
=
true
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
conv2DBackpropInput
(
inputSizes:
Tensor
,
filter:
Tensor
,
outputGradient:
Tensor
,
stride1:
Long
,
stride2:
Long
,
padding:
PaddingMode
,
dataFormat:
CNNDataFormat
=
CNNDataFormat.default
,
useCuDNNOnGPU:
Boolean
=
true
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
crelu
(
input:
Tensor
)
:
Tensor
Definition Classes
NN
def
dropout
(
input:
Tensor
,
keepProbability:
Float
,
noiseShape:
Tensor
=
null
,
seed:
Option
[
Int
] =
None
)
:
Tensor
Definition Classes
NN
def
elu
(
input:
Tensor
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
final
def
eq
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
def
equals
(
arg0:
Any
)
:
Boolean
Definition Classes
AnyRef → Any
def
finalize
()
:
Unit
Attributes
protected[
java.lang
]
Definition Classes
AnyRef
Annotations
@throws
(
classOf[java.lang.Throwable]
)
final
def
getClass
()
:
Class
[_]
Definition Classes
AnyRef → Any
def
hashCode
()
:
Int
Definition Classes
AnyRef → Any
def
inTopK
(
predictions:
Tensor
,
targets:
Tensor
,
k:
Tensor
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
final
def
isInstanceOf
[
T0
]
:
Boolean
Definition Classes
Any
def
l2Loss
(
input:
Tensor
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
l2Normalize
(
x:
Tensor
,
axes:
Tensor
,
epsilon:
Float
=
1e-12f
)
:
Tensor
Definition Classes
NN
def
linear
(
x:
Tensor
,
weights:
Tensor
,
bias:
Tensor
=
null
)
:
Tensor
Definition Classes
NN
def
logPoissonLoss
(
logPredictions:
Tensor
,
targets:
Tensor
,
computeFullLoss:
Boolean
=
false
)
:
Tensor
Definition Classes
NN
def
logSoftmax
(
logits:
Tensor
,
axis:
Int
=
1
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
maxPool
(
input:
Tensor
,
windowSize:
Seq
[
Long
]
,
stride1:
Long
,
stride2:
Long
,
padding:
PaddingMode
,
dataFormat:
CNNDataFormat
=
CNNDataFormat.default
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
maxPoolGrad
(
originalInput:
Tensor
,
originalOutput:
Tensor
,
outputGradient:
Tensor
,
windowSize:
Seq
[
Long
]
,
stride1:
Long
,
stride2:
Long
,
padding:
PaddingMode
,
dataFormat:
CNNDataFormat
=
CNNDataFormat.default
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
maxPoolGradGrad
(
originalInput:
Tensor
,
originalOutput:
Tensor
,
outputGradient:
Tensor
,
windowSize:
Seq
[
Long
]
,
stride1:
Long
,
stride2:
Long
,
padding:
PaddingMode
,
dataFormat:
CNNDataFormat
=
CNNDataFormat.default
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
final
def
ne
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
final
def
notify
()
:
Unit
Definition Classes
AnyRef
final
def
notifyAll
()
:
Unit
Definition Classes
AnyRef
def
relu
(
input:
Tensor
,
alpha:
Float
=
0.0f
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
relu6
(
input:
Tensor
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
selu
(
input:
Tensor
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
sequenceLoss
(
logits:
Tensor
,
labels:
Tensor
,
weights:
Tensor
=
null
,
averageAcrossTimeSteps:
Boolean
=
true
,
averageAcrossBatch:
Boolean
=
true
,
lossFn: (
Tensor
,
Tensor
) ⇒
Tensor
=
sparseSoftmaxCrossEntropy(_, _)
)
:
Tensor
Definition Classes
NN
Annotations
@throws
(
...
)
def
sigmoidCrossEntropy
(
logits:
Tensor
,
labels:
Tensor
,
weights:
Tensor
=
null
)
:
Tensor
Definition Classes
NN
def
softmax
(
logits:
Tensor
,
axis:
Int
=
1
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
softmaxCrossEntropy
(
logits:
Tensor
,
labels:
Tensor
,
axis:
Int
=
1
)
:
Tensor
Definition Classes
NN
def
softplus
(
input:
Tensor
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
softsign
(
input:
Tensor
)
(
implicit
context:
DynamicVariable
[
Context
]
)
:
Tensor
Definition Classes
NN
def
sparseSoftmaxCrossEntropy
(
logits:
Tensor
,
labels:
Tensor
,
axis:
Int
=
1
)
:
Tensor
Definition Classes
NN
final
def
synchronized
[
T0
]
(
arg0: ⇒
T0
)
:
T0
Definition Classes
AnyRef
def
toString
()
:
String
Definition Classes
AnyRef → Any
def
topK
(
input:
Tensor
,
k:
Tensor
=
1
,
sorted:
Boolean
=
true
)
(
implicit
context:
DynamicVariable
[
Context
]
)
: (
Tensor
,
Tensor
)
Definition Classes
NN
final
def
wait
()
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
wait
(
arg0:
Long
,
arg1:
Int
)
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
wait
(
arg0:
Long
)
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
Inherited from
NN
Inherited from
AnyRef
Inherited from
Any
Ungrouped