Skip to content

Commit 0d39b87

Browse files
authored
Merge 340e075 into 62fe166
2 parents 62fe166 + 340e075 commit 0d39b87

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

pyttb/tensor.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2959,8 +2959,7 @@ def tenrand(shape: Shape, order: MemoryLayout = "F") -> tensor:
29592959
# Typing doesn't play nice with partial
29602960
# mypy issue: 1484
29612961
def unit_uniform(pass_through_shape: Tuple[int, ...]) -> np.ndarray:
2962-
data = np.random.uniform(low=0, high=1, size=pass_through_shape)
2963-
to_memory_order(data, order)
2962+
data = np.random.uniform(low=0, high=1, size=np.prod(pass_through_shape))
29642963
return data
29652964

29662965
return tensor.from_function(unit_uniform, shape)

0 commit comments

Comments
 (0)