Class CapsuleUtils


  • public class CapsuleUtils
    extends Object
    • 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)