fix: subtype check bug

This commit is contained in:
Shunsuke Shibayama 2024-02-04 17:29:34 +09:00
parent 3be5d75d05
commit b8fe8fb205
3 changed files with 20 additions and 9 deletions

View file

@ -45,5 +45,11 @@
) => NoneType
# TODO: S bound
item: |T|(self: Ref .Tensor!(T, _)) -> T
to: (|T, S: [Nat; _]|(
self: .Tensor!(T, S),
other: .DType or .Device,
non_blocking := Bool,
copy := Bool,
) -> .Tensor!(T, S))
.relu: |T, S: [Nat; _]|(x: .Tensor!(T, S)) -> .Tensor!(T, S)