Skip to content

Commit d5bd4e5

Browse files
committed
Fix tensor layout
1 parent 1dc4bcd commit d5bd4e5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/0_Tensor/tensor_manipulation.f90

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ program tensor_manipulation
2323
integer(c_int64_t), dimension(2), parameter :: tensor_shape = [2, 3]
2424

2525
! Variables for constructing tensors with torch_tensor_from_array
26-
integer, parameter :: tensor_layout(ndims) = [1, 1]
26+
integer, parameter :: tensor_layout(ndims) = [1, 2]
2727
real(wp), dimension(2,3), target :: in_data
2828

2929
! Array for extracting an array from a tensor

0 commit comments

Comments
 (0)