sigmoid_internal.h File Reference

#include <math.h>
#include "tensor/tensor.h"

Include dependency graph for sigmoid_internal.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
template<typename T >
void	magmadnn::internal::sigmoid_full (Tensor< T > *x, Tensor< T > *out, bool fast)
template<typename T >
void	magmadnn::internal::sigmoid_grad (Tensor< T > *output, Tensor< T > *grad, Tensor< T > *out)

Detailed Description

Author

Daniel Nichols

Version

1.0

Date

2019-02-23

Copyright

Copyright (c) 2019

Function Documentation

sigmoid_full()

template<typename T >

void magmadnn::internal::sigmoid_full	(	Tensor< T > *	x,
		Tensor< T > *	out,
		bool	fast = true
	)

Computes the element-wise sigmoid on x.

Template Parameters

T

Parameters

x	pointer to tensor to be sigmoided
fast	if true, then x=1/(1+|x|) is computed instead of normal sigmoid