mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Add api info for torch._C._nn.pyi (#148405)
APis involved are as followed: - adaptive_avg_pool2d - adaptive_avg_pool3d - binary_cross_entropy - col2im ISSUE Related: https://github.com/pytorch/pytorch/issues/148404 Pull Request resolved: https://github.com/pytorch/pytorch/pull/148405 Approved by: https://github.com/ezyang
This commit is contained in:
parent
7070ab3180
commit
1982ec2d22
|
|
@ -412,6 +412,16 @@ def gen_nn_functional(fm: FileManager) -> None:
|
||||||
"tuple[Tensor, Tensor]",
|
"tuple[Tensor, Tensor]",
|
||||||
)
|
)
|
||||||
],
|
],
|
||||||
|
f"adaptive_avg_pool{d}d": [
|
||||||
|
defs(
|
||||||
|
f"adaptive_avg_pool{d}d",
|
||||||
|
[
|
||||||
|
INPUT,
|
||||||
|
"output_size: _int | _size",
|
||||||
|
],
|
||||||
|
"Tensor",
|
||||||
|
)
|
||||||
|
],
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -516,6 +526,31 @@ def gen_nn_functional(fm: FileManager) -> None:
|
||||||
"Tensor",
|
"Tensor",
|
||||||
)
|
)
|
||||||
],
|
],
|
||||||
|
"binary_cross_entropy": [
|
||||||
|
defs(
|
||||||
|
"binary_cross_entropy",
|
||||||
|
[
|
||||||
|
INPUT,
|
||||||
|
"target: Tensor",
|
||||||
|
"weight: Tensor | None = None",
|
||||||
|
"reduction: str = ...",
|
||||||
|
],
|
||||||
|
"Tensor",
|
||||||
|
)
|
||||||
|
],
|
||||||
|
"col2im": [
|
||||||
|
defs(
|
||||||
|
"col2im",
|
||||||
|
[
|
||||||
|
INPUT,
|
||||||
|
"output_size: _int | _size",
|
||||||
|
KERNEL_SIZE,
|
||||||
|
"dilation: _int | _size",
|
||||||
|
*STRIDE_PADDING,
|
||||||
|
],
|
||||||
|
"Tensor",
|
||||||
|
)
|
||||||
|
],
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user