|
|
Find the shape into which shape1 and shape2 can be expanded
Parameter Types:
-
shape1
:
Shape
-
shape2
:
Shape
Return Type:
int []
|
|
|
Find the shape into which all the shapes can be expanded
Parameter Types:
Return Type:
Shape
|
canExpand oldShape newShape
|
Parameter Types:
-
oldShape
:
Shape
-
newShape
:
Shape
Return Type:
bool
|
checkCanAddSlice shape1 location shape2
|
Parameter Types:
-
shape1
:
Shape
-
location
:
int []
-
shape2
:
Shape
|
|
|
Parameter Types:
-
shapes
:
Shape []
-
dim
:
int
Return Type:
int * int [] * int * int [] * int []
|
checkCanConv1d dtype1 dtype2 shape1 shape2 stride padding dilation
|
Parameter Types:
-
dtype1
:
Dtype
-
dtype2
:
Dtype
-
shape1
:
Shape
-
shape2
:
Shape
-
stride
:
int
-
padding
:
int
-
dilation
:
int
Return Type:
int * int * int * int * int * int []
|
checkCanConv2d dtype1 dtype2 shape1 shape2 stride padding dilation
|
Parameter Types:
-
dtype1
:
Dtype
-
dtype2
:
Dtype
-
shape1
:
Shape
-
shape2
:
Shape
-
stride
:
int []
-
padding
:
int []
-
dilation
:
int []
Return Type:
int * int * (int * int) * (int * int * int) * int []
|
checkCanConv3d dtype1 dtype2 shape1 shape2 stride padding dilation
|
Parameter Types:
-
dtype1
:
Dtype
-
dtype2
:
Dtype
-
shape1
:
Shape
-
shape2
:
Shape
-
stride
:
int []
-
padding
:
int []
-
dilation
:
int []
Return Type:
int * int * (int * int * int) * (int * int * int * int) * int []
|
checkCanDilate dim dilations
|
Parameter Types:
-
dim
:
int
-
dilations
:
int []
|
checkCanDot shape1 shape2
|
Parameter Types:
-
shape1
:
Shape
-
shape2
:
Shape
|
|
|
Parameter Types:
|
checkCanDropout2d shape p
|
Parameter Types:
|
checkCanDropout3d shape p
|
Parameter Types:
|
checkCanExpand oldShape newShape
|
Parameter Types:
-
oldShape
:
Shape
-
newShape
:
Shape
|
checkCanFlatten shape startDim endDim
|
Parameter Types:
-
shape
:
Shape
-
startDim
:
int
-
endDim
:
int
|
|
|
Parameter Types:
|
checkCanGather shape dim indicesShape indicesDtype
|
Parameter Types:
-
shape
:
Shape
-
dim
:
int
-
indicesShape
:
Shape
-
indicesDtype
:
Dtype
|
checkCanGetSlice shape fullBounds
|
Parameter Types:
-
shape
:
Shape
-
fullBounds
:
int [,]
Return Type:
int []
|
checkCanMatmul shape1 shape2
|
Parameter Types:
-
shape1
:
Shape
-
shape2
:
Shape
|
checkCanMaxpool1d shape kernelSize stride padding
|
Parameter Types:
-
shape
:
Shape
-
kernelSize
:
int
-
stride
:
int
-
padding
:
int
Return Type:
int * int * int * int * int []
|
checkCanMaxpool2d shape kernelSize stride padding
|
Parameter Types:
-
shape
:
Shape
-
kernelSize
:
int []
-
stride
:
int []
-
padding
:
int []
Return Type:
int * int * (int * int) * (int * int) * (int * int) * int []
|
checkCanMaxpool3d shape kernelSize stride padding
|
Parameter Types:
-
shape
:
Shape
-
kernelSize
:
int []
-
stride
:
int []
-
padding
:
int []
Return Type:
int * int * (int * int * int) * (int * int * int) * (int * int * int) * int []
|
checkCanMaxunpool1d shape indicesDtype indicesShape outputSize
|
Parameter Types:
-
shape
:
Shape
-
indicesDtype
:
Dtype
-
indicesShape
:
Shape
-
outputSize
:
int []
Return Type:
int * int * int * int []
|
checkCanMaxunpool2d shape indicesDtype indicesShape outputSize
|
Parameter Types:
-
shape
:
Shape
-
indicesDtype
:
Dtype
-
indicesShape
:
Shape
-
outputSize
:
int []
Return Type:
int * int * (int * int) * int []
|
checkCanMaxunpool3d shape indicesDtype indicesShape outputSize
|
Parameter Types:
-
shape
:
Shape
-
indicesDtype
:
Dtype
-
indicesShape
:
Shape
-
outputSize
:
int []
Return Type:
int * int * (int * int * int) * int []
|
checkCanPad shape paddings
|
Parameter Types:
-
shape
:
Shape
-
paddings
:
int []
|
|
|
Parameter Types:
|
checkCanSplit shape sizes dim
|
Parameter Types:
-
shape
:
Shape
-
sizes
:
int []
-
dim
:
int
Return Type:
int [] []
|
|
|
Parameter Types:
-
shapes
:
Shape []
-
dim
:
int
Return Type:
int * int [] * int [] * int []
|
checkCanTranspose shape dim0 dim1
|
Parameter Types:
-
shape
:
Shape
-
dim0
:
int
-
dim1
:
int
|
|
|
Parameter Types:
|
checkCanUnsqueeze dim shape
|
Parameter Types:
Return Type:
int []
|
checkCanUnstack shape dim
|
Parameter Types:
Return Type:
int [] * int [] * int []
|
checkCanView shape1 shape2
|
Parameter Types:
-
shape1
:
Shape
-
shape2
:
Shape
|
|
|
Parameter Types:
-
nelement
:
int
-
shape
:
Shape
Return Type:
Shape
|
|
|
Parameter Types:
Return Type:
int
|
|
|
Parameter Types:
Return Type:
int []
|
contains bigShape smallShape
|
Parameter Types:
-
bigShape
:
Shape
-
smallShape
:
Shape
Return Type:
bool
|
|
|
Parameter Types:
-
shape
:
Shape
-
dilations
:
int []
Return Type:
int []
|
|
|
Parameter Types:
-
shape
:
Shape
-
dilations
:
int []
Return Type:
int []
|
flatten startDim endDim shape
|
Parameter Types:
-
startDim
:
int
-
endDim
:
int
-
shape
:
Shape
Return Type:
int []
|
locationToBounds shape location
|
Parameter Types:
-
shape
:
Shape
-
location
:
int []
Return Type:
int [,]
|
scalar
Full Usage:
scalar
Return Type:
Shape
Signature:
Shape
|
Return Type:
Shape
|
|
|
Parameter Types:
Return Type:
int []
|
undilatedShape shape dilations
|
Parameter Types:
-
shape
:
Shape
-
dilations
:
int []
Return Type:
int []
|
unsqueezeAs shape1 shape2
|
Parameter Types:
-
shape1
:
Shape
-
shape2
:
Shape
Return Type:
int []
|