functional

Documentation

Index

• func AdaptiveAvgPool2d(input torch.Tensor, outputSize []int64) torch.Tensor
• func BatchNorm(input, runningMean, runningVar, weight, bias torch.Tensor, training bool, ...) torch.Tensor
• func BinaryCrossEntropy(input, target, weight torch.Tensor, reduction string) torch.Tensor
• func Conv2d(input, weight, bias torch.Tensor, stride, padding, dilation []int64, ...) torch.Tensor
• func ConvTranspose2d(input, weight, bias torch.Tensor, stride, padding, outputPadding []int64, ...) torch.Tensor
• func CrossEntropy(input, target, weight torch.Tensor, ignoreIndex int64, reduction string) torch.Tensor
• func LeakyRelu(input torch.Tensor, negativeSlope float64, inplace bool) torch.Tensor
• func Linear(input, weight, bias torch.Tensor) torch.Tensor
• func LogSoftmax(input torch.Tensor, dim int64) torch.Tensor
• func MaxPool2d(input torch.Tensor, kernelSize, stride, padding, dilation []int64, ...) torch.Tensor
• func NllLoss(input, target, weight torch.Tensor, ignoreIndex int64, reduction string) torch.Tensor
• func Relu(input torch.Tensor, inplace bool) torch.Tensor

Functions

func AdaptiveAvgPool2d

func AdaptiveAvgPool2d(input torch.Tensor, outputSize []int64) torch.Tensor

AdaptiveAvgPool2d torch.nn.functional.adaptive_avg_pool2d

func BatchNorm

func BatchNorm(input, runningMean, runningVar, weight, bias torch.Tensor,
	training bool, momentum, eps float64) torch.Tensor

BatchNorm does batch nomalization for `input`

func BinaryCrossEntropy

func BinaryCrossEntropy(input, target, weight torch.Tensor,
	reduction string) torch.Tensor

BinaryCrossEntropy torch.nn.functional.binary_cross_entropy

func Conv2d

func Conv2d(input, weight, bias torch.Tensor,
	stride, padding, dilation []int64, groups int64) torch.Tensor

Conv2d does 2d-convolution

func ConvTranspose2d

func ConvTranspose2d(
	input, weight, bias torch.Tensor,
	stride, padding, outputPadding []int64,
	groups int64, dilation []int64) torch.Tensor

ConvTranspose2d does 2d-fractionally-strided convolution

func CrossEntropy

func CrossEntropy(input, target, weight torch.Tensor, ignoreIndex int64,
	reduction string) torch.Tensor

CrossEntropy torch.nn.functional.cross_entropy

func LeakyRelu

func LeakyRelu(input torch.Tensor, negativeSlope float64, inplace bool) torch.Tensor

LeakyRelu torch.nn.functional.leaky_relu

func Linear

func Linear(input, weight, bias torch.Tensor) torch.Tensor

Linear ports torch.nn.functional.linear

func LogSoftmax

func LogSoftmax(input torch.Tensor, dim int64) torch.Tensor

LogSoftmax torch.nn.functional.log_softmax

func MaxPool2d

func MaxPool2d(input torch.Tensor, kernelSize, stride, padding,
	dilation []int64, ceilMode bool) torch.Tensor

MaxPool2d torch.nn.functional.max_pool2d

func NllLoss

func NllLoss(input, target, weight torch.Tensor, ignoreIndex int64,
	reduction string) torch.Tensor

NllLoss torch.nn.functional.nll_loss

func Relu

func Relu(input torch.Tensor, inplace bool) torch.Tensor

Relu torch.nn.functional.relu