org.deeplearning4j.util

Class CapsuleUtils

java.lang.Object

org.deeplearning4j.util.CapsuleUtils

public class CapsuleUtils
extends Object

Utilities for CapsNet Layers

Author:

Ryan Nett

See Also:

CapsuleLayer, PrimaryCapsules, CapsuleStrengthLayer

Constructor Summary

Constructors

Constructor and Description
CapsuleUtils()

Method Summary

Modifier and Type	Method and Description
static SDVariable	softmax(SameDiff SD, SDVariable x, int dimension, int rank) Compute softmax along a given dimension
static SDVariable	squash(SameDiff SD, SDVariable x, int dim) Compute the squash operation used in CapsNet The formula is (||s||^2 / (1 + ||s||^2)) * (s / ||s||).

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

CapsuleUtils

public CapsuleUtils()

Method Detail

squash

public static SDVariable squash(SameDiff SD,
                                SDVariable x,
                                int dim)

Compute the squash operation used in CapsNet The formula is (||s||^2 / (1 + ||s||^2)) * (s / ||s||). Canceling one ||s|| gives ||s||*s/((1 + ||s||^2)

Parameters:

SD - The SameDiff environment

x - The variable to squash

Returns:

squash(x)

softmax

public static SDVariable softmax(SameDiff SD,
                                 SDVariable x,
                                 int dimension,
                                 int rank)

Compute softmax along a given dimension