Documentation
¶
Index ¶
- func InterpolateScale(input torch.Tensor, scale []float64, mode InterpolateMode, alignCorners bool, ...) torch.Tensor
- func InterpolateSize(input torch.Tensor, size []int64, mode InterpolateMode, alignCorners bool, ...) torch.Tensor
- func Normalize(input torch.Tensor, p float64, dim int, eps float64) torch.Tensor
- func Pad(input torch.Tensor, padding []int64, mode PadMode, value ...float64) torch.Tensor
- func Relu(tensor torch.Tensor, inplace bool) torch.Tensor
- func Softmax(tensor torch.Tensor, dim int64) torch.Tensor
- type InterpolateMode
- type PadMode
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func InterpolateScale ¶
func InterpolateScale( input torch.Tensor, scale []float64, mode InterpolateMode, alignCorners bool, antialias bool, ) torch.Tensor
Interpolate a tensor by scale.
func InterpolateSize ¶
func InterpolateSize( input torch.Tensor, size []int64, mode InterpolateMode, alignCorners bool, antialias bool, ) torch.Tensor
Interpolate a tensor by size.
Types ¶
type InterpolateMode ¶
type InterpolateMode int64
Interpolation algorithms implemented by libtorch.
const ( InterpolateNearest InterpolateMode = iota InterpolateLinear InterpolateBilinear InterpolateBicubic InterpolateTrilinear InterpolateArea InterpolateNearestExact )
Source Files
Click to show internal directories.
Click to hide internal directories.