Commit 1c45389
authored
[BugFix] Fixed Inappropriate Logical Expression (#16272)
[BugFix] Fixed a comparison for splitting tensor
In the `tensor_split` method, there's a comparsion
that checks if the input tensor is zero-dimensional
or one-dimensional long tensor.
In the comparsion, there's a typo that converts
the shape of the tensor to a list and compares
against integer.
This commit fixes the bug by comapring the
length of the tensor against the integer.
Signed-off-by: fazledyn-or <ataf@openrefactory.com>1 parent a050696 commit 1c45389
1 file changed
Lines changed: 1 addition & 3 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
595 | 595 | | |
596 | 596 | | |
597 | 597 | | |
598 | | - | |
599 | | - | |
600 | | - | |
| 598 | + | |
601 | 599 | | |
602 | 600 | | |
603 | 601 | | |
| |||
0 commit comments