pytorch Namespace Reference

Namespaces
	internal

Classes
class	AvgPool2d
class	BatchNorm2d
class	Branch
class	Concat
class	Conv2d
class	Difference
class	Divisor
class	Hardtanh
class	inference_engine
class	Layer
class	Linear
class	MaxPool2d
class	MaxUnpool2d
class	Product
class	ReLU
class	RNN
class	Sigmoid
class	Skip
class	Slice
class	Softmax
class	Sum
class	Tanh

Enumerations
enum	dims { n = 3, k = 2, h = 0, w = 1 }
	Convenience enum to use whenever you need to specify a dimension (like in Concat) More...

Enumeration Type Documentation

dims

enum pytorch::dims

Convenience enum to use whenever you need to specify a dimension (like in Concat)

Enumerator
n
k
h
w