Skip to content

others

Module: others.py

This module implements other helpful functions for Neural Networks

Authors
  • Lokesh Mohanty (lokeshm@iisc.ac.in)
Version Info
  • 01/01/2025: Initial version
  • 06/01/2025: Rename and add other functions from jax.nn

log_softmax(x)

Compute log_softmax activation

Parameters:

Name Type Description Default
x Array

input data

 required
Source code in scirex/core/dl/nn/others.py 
51
52
53
54
55
56
57
58
def log_softmax(x: jax.Array) -> jax.Array:
    """
    Compute log_softmax activation

    Args:
        x: input data
    """
    return jax.nn.log_softmax(x)

one_hot(x)

Compute one_hot activation

Parameters:

Name Type Description Default
x Array

input data

 required
Source code in scirex/core/dl/nn/others.py 
71
72
73
74
75
76
77
78
def one_hot(x: jax.Array) -> jax.Array:
    """
    Compute one_hot activation

    Args:
        x: input data
    """
    return jax.nn.one_hot(x)

softmax(x)

Compute softmax activation

Parameters:

Name Type Description Default
x Array

input data

 required
Source code in scirex/core/dl/nn/others.py 
41
42
43
44
45
46
47
48
def softmax(x: jax.Array) -> jax.Array:
    """
    Compute softmax activation

    Args:
        x: input data
    """
    return jax.nn.softmax(x)

standardize(x)

Compute standardize activation

Parameters:

Name Type Description Default
x Array

input data

 required
Source code in scirex/core/dl/nn/others.py 
61
62
63
64
65
66
67
68
def standardize(x: jax.Array) -> jax.Array:
    """
    Compute standardize activation

    Args:
        x: input data
    """
    return jax.nn.standardize(x)